{ | |
"args": [], | |
"kwargs": { | |
"intermediate_size": [ | |
448, | |
448 | |
], | |
"threshold": 0.2, | |
"max_overlap": 0.25, | |
"grid_size": [ | |
7, | |
7 | |
], | |
"class_map": { | |
"aeroplane": 0, | |
"bicycle": 1, | |
"bird": 2, | |
"boat": 3, | |
"bottle": 4, | |
"bus": 5, | |
"car": 6, | |
"cat": 7, | |
"chair": 8, | |
"cow": 9, | |
"diningtable": 10, | |
"dog": 11, | |
"horse": 12, | |
"motorbike": 13, | |
"person": 14, | |
"pottedplant": 15, | |
"sheep": 16, | |
"sofa": 17, | |
"train": 18, | |
"tvmonitor": 19 | |
} | |
}, | |
"class_name": "YoloV1Pipeline", | |
"module_name": "vision_models_playground.pipelines.yolo" | |
} |