StephanST's picture
adding better trained model
4f5e042
raw
history blame
929 Bytes
{
"version": "1.1.0",
"codebase": {
"task": "Segmentation",
"codebase": "mmseg",
"version": "1.0.0",
"pth": "../mmsegmentation/work_dirs/mobilenet_deeplab_drone/iter_96000.pth",
"config": "../mmsegmentation/work_dirs/mobilenet_deeplab_drone/mobilenet_deeplab_drone.py"
},
"codebase_config": {
"type": "mmseg",
"task": "Segmentation",
"with_argmax": true
},
"onnx_config": {
"type": "onnx",
"export_params": true,
"keep_initializers_as_inputs": false,
"opset_version": 11,
"save_file": "end2end.onnx",
"input_names": [
"input"
],
"output_names": [
"output"
],
"input_shape": [
416,
416
],
"optimize": true
},
"backend_config": {
"type": "onnxruntime"
},
"calib_config": {}
}