| { |
| "_name_or_path": "davidhajdu/dab-detr-resnet-50", |
| "activation_dropout": 0.0, |
| "activation_function": "prelu", |
| "architectures": [ |
| "DabDetrForObjectDetection" |
| ], |
| "attention_dropout": 0.0, |
| "auxiliary_loss": false, |
| "backbone": "resnet50", |
| "backbone_config": null, |
| "backbone_kwargs": { |
| "in_chans": 3, |
| "out_indices": [ |
| 1, |
| 2, |
| 3, |
| 4 |
| ] |
| }, |
| "bbox_cost": 5, |
| "bbox_loss_coefficient": 5, |
| "class_cost": 2, |
| "cls_loss_coefficient": 2, |
| "decoder_attention_heads": 8, |
| "decoder_ffn_dim": 2048, |
| "decoder_layers": 6, |
| "dropout": 0.1, |
| "encoder_attention_heads": 8, |
| "encoder_ffn_dim": 2048, |
| "encoder_layers": 6, |
| "focal_alpha": 0.25, |
| "giou_cost": 2, |
| "giou_loss_coefficient": 2, |
| "hidden_size": 256, |
| "id2label": { |
| "0": "Coverall", |
| "1": "Face_Shield", |
| "2": "Gloves", |
| "3": "Goggles", |
| "4": "Mask" |
| }, |
| "init_std": 0.02, |
| "init_xavier_std": 1.0, |
| "initializer_bias_prior_prob": null, |
| "is_encoder_decoder": true, |
| "keep_query_pos": false, |
| "label2id": { |
| "Coverall": 0, |
| "Face_Shield": 1, |
| "Gloves": 2, |
| "Goggles": 3, |
| "Mask": 4 |
| }, |
| "model_type": "dab-detr", |
| "normalize_before": false, |
| "num_hidden_layers": 6, |
| "num_patterns": 0, |
| "num_queries": 300, |
| "query_dim": 4, |
| "random_refpoints_xy": false, |
| "sine_position_embedding_scale": null, |
| "temperature_height": 20, |
| "temperature_width": 20, |
| "torch_dtype": "float32", |
| "transformers_version": "4.47.0.dev0", |
| "use_pretrained_backbone": true, |
| "use_timm_backbone": true |
| } |
|
|
