| { | |
| "amp": true, | |
| "architectures": [ | |
| "RFDetrModelForObjectDetection" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_rf_detr.RFDetrConfig", | |
| "AutoModelForObjectDetection": "modeling_rf_detr.RFDetrModelForObjectDetection" | |
| }, | |
| "bbox_reparam": true, | |
| "ca_nheads": 16, | |
| "dec_layers": 3, | |
| "dec_n_points": 2, | |
| "device": "cpu", | |
| "encoder": "dinov2_windowed_small", | |
| "gradient_checkpointing": false, | |
| "group_detr": 13, | |
| "hidden_dim": 256, | |
| "layer_norm": true, | |
| "lite_refpoint_refine": true, | |
| "model_name": "RFDETRBase", | |
| "model_type": "rf-detr", | |
| "num_classes": 90, | |
| "num_queries": 300, | |
| "out_feature_indexes": [ | |
| 2, | |
| 5, | |
| 8, | |
| 11 | |
| ], | |
| "pretrain_weights": "rf-detr-base.pth", | |
| "pretrained": true, | |
| "projector_scale": [ | |
| "P4" | |
| ], | |
| "resolution": 560, | |
| "sa_nheads": 8, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.50.3", | |
| "two_stage": true | |
| } | |