32 lines
1.0 KiB
JSON
32 lines
1.0 KiB
JSON
{
|
|
"model" : {
|
|
"type": "SegNet",
|
|
"architecture": "MobileNet7_5",
|
|
"input_size": 224,
|
|
"n_classes" : 20
|
|
},
|
|
"weights" : {
|
|
"full": "",
|
|
"backend": "imagenet"
|
|
},
|
|
"train" : {
|
|
"actual_epoch": 1,
|
|
"train_image_folder": "sample_datasets/segmentation/imgs",
|
|
"train_annot_folder": "sample_datasets/segmentation/anns",
|
|
"train_times": 4,
|
|
"valid_image_folder": "sample_datasets/segmentation/imgs_validation",
|
|
"valid_annot_folder": "sample_datasets/segmentation/anns_validation",
|
|
"valid_times": 4,
|
|
"valid_metric": "val_loss",
|
|
"batch_size": 8,
|
|
"learning_rate": 1e-4,
|
|
"saved_folder": "segment",
|
|
"first_trainable_layer": "",
|
|
"ignore_zero_class": false,
|
|
"augmentation": true
|
|
},
|
|
"converter" : {
|
|
"type": ["k210", "tflite"]
|
|
}
|
|
}
|