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