File size: 524 Bytes
c5388a3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | {
"model_type": "dfine",
"architectures": ["DFINEModel"],
"auto_map": {
"AutoConfig": "modeling_dfine.DFINEConfig",
"AutoModel": "modeling_dfine.DFINEModel"
},
"num_classes": 4,
"id2label": {
"0": "Object-detection",
"1": "Figure",
"2": "Icon",
"3": "Table"
},
"label2id": {
"Object-detection": 0,
"Figure": 1,
"Icon": 2,
"Table": 3
},
"resolution": 640,
"checkpoint": "best_stg2.pth",
"config_file": "dfine_hgnetv2_x_custom.yml",
"backbone": "HGNetv2-B5"
}
|