Zin / basketviewer Public
Primary version

Training settings

Please provide a valid number of training cycles (numeric only)
Please provide a valid number for the learning rate (between 0 and 1)
Please provide a valid number for the train/validate split (between 0 and 1)

Neural network architecture

sys.path.append('./resources/libraries') import ei_tensorflow.object_detection EPOCHS = args.epochs or 100 LEARNING_RATE = args.learning_rate or 0.05 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)
Output layer (2 classes)