Training settings
Please provide a valid training processor option
Neural network architecture
sys.path.append('./resources/libraries')
import ei_tensorflow.object_detection
EPOCHS = args.epochs or 50
LEARNING_RATE = args.learning_rate or 0.005
ei_tensorflow.object_detection.train(classes, LEARNING_RATE, EPOCHS,
train_dataset, validation_dataset)
# The 'model' variable is expected to be a Keras model. For this model type there is no keras model,
# so we just set it to None.
model = None
Input layer (307,200 features)
MobileNetV2 SSD FPN-Lite 320x320
Output layer (2 classes)
Model
Model version: