33 lines
1018 B
JSON
33 lines
1018 B
JSON
{
|
|
"model" : {
|
|
"type": "Classifier",
|
|
"architecture": "MobileNet7_5",
|
|
"input_size": 224,
|
|
"fully-connected": [100,50],
|
|
"labels": [],
|
|
"dropout" : 0.5
|
|
},
|
|
"weights" : {
|
|
"full": "",
|
|
"backend": "imagenet",
|
|
"save_bottleneck": false
|
|
|
|
},
|
|
"train" : {
|
|
"actual_epoch": 1,
|
|
"train_image_folder": "sample_datasets/classifier/imgs",
|
|
"train_times": 4,
|
|
"valid_image_folder": "sample_datasets/classifier/imgs_validation",
|
|
"valid_times": 4,
|
|
"valid_metric": "val_accuracy",
|
|
"batch_size": 4,
|
|
"learning_rate": 1e-4,
|
|
"saved_folder": "classifier",
|
|
"first_trainable_layer": "",
|
|
"augmentation": true
|
|
},
|
|
"converter" : {
|
|
"type": ["k210","tflite"]
|
|
}
|
|
}
|