CoolFace
Modelpublic

gobk/cppe5_use_data_finetuning

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes10downloads
config.json61 linesDownload Raw Back to root
1{2  "_name_or_path": "facebook/detr-resnet-50",3  "activation_dropout": 0.0,4  "activation_function": "relu",5  "architectures": [6    "DetrForObjectDetection"7  ],8  "attention_dropout": 0.0,9  "auxiliary_loss": false,10  "backbone": "resnet50",11  "backbone_config": null,12  "bbox_cost": 5,13  "bbox_loss_coefficient": 5,14  "class_cost": 1,15  "classifier_dropout": 0.0,16  "d_model": 256,17  "decoder_attention_heads": 8,18  "decoder_ffn_dim": 2048,19  "decoder_layerdrop": 0.0,20  "decoder_layers": 6,21  "dice_loss_coefficient": 1,22  "dilation": false,23  "dropout": 0.1,24  "encoder_attention_heads": 8,25  "encoder_ffn_dim": 2048,26  "encoder_layerdrop": 0.0,27  "encoder_layers": 6,28  "eos_coefficient": 0.1,29  "giou_cost": 2,30  "giou_loss_coefficient": 2,31  "id2label": {32    "0": "Coverall",33    "1": "Face_Shield",34    "2": "Gloves",35    "3": "Goggles",36    "4": "Mask"37  },38  "init_std": 0.02,39  "init_xavier_std": 1.0,40  "is_encoder_decoder": true,41  "label2id": {42    "Coverall": 0,43    "Face_Shield": 1,44    "Gloves": 2,45    "Goggles": 3,46    "Mask": 447  },48  "mask_loss_coefficient": 1,49  "max_position_embeddings": 1024,50  "model_type": "detr",51  "num_channels": 3,52  "num_hidden_layers": 6,53  "num_queries": 100,54  "position_embedding_type": "sine",55  "scale_embedding": false,56  "torch_dtype": "float32",57  "transformers_version": "4.34.1",58  "use_pretrained_backbone": true,59  "use_timm_backbone": true60}61