pipeline/config.json

{
    "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"
}