| { | |
| "model_type": "yolov8", | |
| "num_labels": 2, | |
| "image_size": 640, | |
| "backbone": "CSPDarknet53", | |
| "pretrained": true, | |
| "input_channels": 3, | |
| "conf_thresh": 0.5, | |
| "iou_thresh": 0.4, | |
| "max_detections": 100, | |
| "stride": [8, 16, 32], | |
| "anchor_generator": { | |
| "sizes": [[12, 16, 32, 64, 128], [256, 512, 1024]], | |
| "aspect_ratios": [[1, 2, 0.5], [1, 2, 0.5]] | |
| }, | |
| "nms_thresh": 0.4 | |
| } | |