CoolFace
Modelpublic

SenseTime/deformable-detr

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
22likes23kdownloads
config.json222 linesDownload Raw Back to root
1{2  "activation_dropout": 0.0,3  "activation_function": "relu",4  "architectures": [5    "DeformableDetrForObjectDetection"6  ],7  "attention_dropout": 0.0,8  "auxiliary_loss": false,9  "backbone": "resnet50",10  "bbox_cost": 5,11  "bbox_loss_coefficient": 5,12  "class_cost": 1,13  "d_model": 256,14  "decoder_attention_heads": 8,15  "decoder_ffn_dim": 1024,16  "decoder_layerdrop": 0.0,17  "decoder_layers": 6,18  "decoder_n_points": 4,19  "dice_loss_coefficient": 1,20  "dilation": false,21  "dropout": 0.1,22  "encoder_attention_heads": 8,23  "encoder_ffn_dim": 1024,24  "encoder_layerdrop": 0.0,25  "encoder_layers": 6,26  "encoder_n_points": 4,27  "eos_coefficient": 0.1,28  "giou_cost": 2,29  "giou_loss_coefficient": 2,30  "id2label": {31    "0": "N/A",32    "1": "person",33    "2": "bicycle",34    "3": "car",35    "4": "motorcycle",36    "5": "airplane",37    "6": "bus",38    "7": "train",39    "8": "truck",40    "9": "boat",41    "10": "traffic light",42    "11": "fire hydrant",43    "12": "N/A",44    "13": "stop sign",45    "14": "parking meter",46    "15": "bench",47    "16": "bird",48    "17": "cat",49    "18": "dog",50    "19": "horse",51    "20": "sheep",52    "21": "cow",53    "22": "elephant",54    "23": "bear",55    "24": "zebra",56    "25": "giraffe",57    "26": "N/A",58    "27": "backpack",59    "28": "umbrella",60    "29": "N/A",61    "30": "N/A",62    "31": "handbag",63    "32": "tie",64    "33": "suitcase",65    "34": "frisbee",66    "35": "skis",67    "36": "snowboard",68    "37": "sports ball",69    "38": "kite",70    "39": "baseball bat",71    "40": "baseball glove",72    "41": "skateboard",73    "42": "surfboard",74    "43": "tennis racket",75    "44": "bottle",76    "45": "N/A",77    "46": "wine glass",78    "47": "cup",79    "48": "fork",80    "49": "knife",81    "50": "spoon",82    "51": "bowl",83    "52": "banana",84    "53": "apple",85    "54": "sandwich",86    "55": "orange",87    "56": "broccoli",88    "57": "carrot",89    "58": "hot dog",90    "59": "pizza",91    "60": "donut",92    "61": "cake",93    "62": "chair",94    "63": "couch",95    "64": "potted plant",96    "65": "bed",97    "66": "N/A",98    "67": "dining table",99    "68": "N/A",100    "69": "N/A",101    "70": "toilet",102    "71": "N/A",103    "72": "tv",104    "73": "laptop",105    "74": "mouse",106    "75": "remote",107    "76": "keyboard",108    "77": "cell phone",109    "78": "microwave",110    "79": "oven",111    "80": "toaster",112    "81": "sink",113    "82": "refrigerator",114    "83": "N/A",115    "84": "book",116    "85": "clock",117    "86": "vase",118    "87": "scissors",119    "88": "teddy bear",120    "89": "hair drier",121    "90": "toothbrush"122  },123  "init_std": 0.02,124  "init_xavier_std": 1.0,125  "is_encoder_decoder": true,126  "label2id": {127    "N/A": 83,128    "airplane": 5,129    "apple": 53,130    "backpack": 27,131    "banana": 52,132    "baseball bat": 39,133    "baseball glove": 40,134    "bear": 23,135    "bed": 65,136    "bench": 15,137    "bicycle": 2,138    "bird": 16,139    "boat": 9,140    "book": 84,141    "bottle": 44,142    "bowl": 51,143    "broccoli": 56,144    "bus": 6,145    "cake": 61,146    "car": 3,147    "carrot": 57,148    "cat": 17,149    "cell phone": 77,150    "chair": 62,151    "clock": 85,152    "couch": 63,153    "cow": 21,154    "cup": 47,155    "dining table": 67,156    "dog": 18,157    "donut": 60,158    "elephant": 22,159    "fire hydrant": 11,160    "fork": 48,161    "frisbee": 34,162    "giraffe": 25,163    "hair drier": 89,164    "handbag": 31,165    "horse": 19,166    "hot dog": 58,167    "keyboard": 76,168    "kite": 38,169    "knife": 49,170    "laptop": 73,171    "microwave": 78,172    "motorcycle": 4,173    "mouse": 74,174    "orange": 55,175    "oven": 79,176    "parking meter": 14,177    "person": 1,178    "pizza": 59,179    "potted plant": 64,180    "refrigerator": 82,181    "remote": 75,182    "sandwich": 54,183    "scissors": 87,184    "sheep": 20,185    "sink": 81,186    "skateboard": 41,187    "skis": 35,188    "snowboard": 36,189    "spoon": 50,190    "sports ball": 37,191    "stop sign": 13,192    "suitcase": 33,193    "surfboard": 42,194    "teddy bear": 88,195    "tennis racket": 43,196    "tie": 32,197    "toaster": 80,198    "toilet": 70,199    "toothbrush": 90,200    "traffic light": 10,201    "train": 7,202    "truck": 8,203    "tv": 72,204    "umbrella": 28,205    "vase": 86,206    "wine glass": 46,207    "zebra": 24208  },209  "mask_loss_coefficient": 1,210  "max_position_embeddings": 1024,211  "model_type": "deformable_detr",212  "num_feature_levels": 4,213  "num_queries": 300,214  "position_embedding_type": "sine",215  "return_intermediate": true,216  "torch_dtype": "float32",217  "transformers_version": "4.16.0.dev0",218  "two_stage": false,219  "two_stage_num_proposals": 300,220  "with_box_refine": false221}222