{ "activation_dropout": 0.0, "activation_function": "relu", "architectures": [ "RelationDetrForObjectDetection" ], "attention_dropout": 0.0, "backbone": "resnet50", "backbone_config": null, "backbone_kwargs": { "in_chans": 3, "out_indices": [ 2, 3, 4 ] }, "bbox_loss_coefficient": 5, "box_noise_scale": 1.0, "class_loss_coefficient": 1, "cost_bbox": 5, "cost_class": 2, "cost_giou": 2, "d_model": 256, "decoder_attention_heads": 8, "decoder_ffn_dim": 2048, "decoder_layers": 6, "decoder_n_points": 4, "denoising_nums": 100, "dilation": false, "disable_custom_kernels": false, "dropout": 0.0, "encoder_attention_heads": 8, "encoder_ffn_dim": 2048, "encoder_layers": 6, "encoder_n_points": 4, "focal_alpha": 0.25, "focal_gamma": 2.0, "giou_loss_coefficient": 2, "hybrid_assign": 6, "hybrid_queries": 1500, "id2label": { "0": "none", "1": "person", "2": "bicycle", "3": "car", "4": "motorcycle", "5": "airplane", "6": "bus", "7": "train", "8": "truck", "9": "boat", "10": "traffic light", "11": "fire hydrant", "12": "none", "13": "stop sign", "14": "parking meter", "15": "bench", "16": "bird", "17": "cat", "18": "dog", "19": "horse", "20": "sheep", "21": "cow", "22": "elephant", "23": "bear", "24": "zebra", "25": "giraffe", "26": "none", "27": "backpack", "28": "umbrella", "29": "none", "30": "none", "31": "handbag", "32": "tie", "33": "suitcase", "34": "frisbee", "35": "skis", "36": "snowboard", "37": "sports ball", "38": "kite", "39": "baseball bat", "40": "baseball glove", "41": "skateboard", "42": "surfboard", "43": "tennis racket", "44": "bottle", "45": "none", "46": "wine glass", "47": "cup", "48": "fork", "49": "knife", "50": "spoon", "51": "bowl", "52": "banana", "53": "apple", "54": "sandwich", "55": "orange", "56": "broccoli", "57": "carrot", "58": "hot dog", "59": "pizza", "60": "donut", "61": "cake", "62": "chair", "63": "couch", "64": "potted plant", "65": "bed", "66": "none", "67": "dining table", "68": "none", "69": "none", "70": "toilet", "71": "none", "72": "tv", "73": "laptop", "74": "mouse", "75": "remote", "76": "keyboard", "77": "cell phone", "78": "microwave", "79": "oven", "80": "toaster", "81": "sink", "82": "refrigerator", "83": "none", "84": "book", "85": "clock", "86": "vase", "87": "scissors", "88": "teddy bear", "89": "hair drier", "90": "toothbrush" }, "init_std": 0.02, "init_xavier_std": 1.0, "label2id": { "LABEL_0": 0, "LABEL_1": 1, "LABEL_10": 10, "LABEL_11": 11, "LABEL_12": 12, "LABEL_13": 13, "LABEL_14": 14, "LABEL_15": 15, "LABEL_16": 16, "LABEL_17": 17, "LABEL_18": 18, "LABEL_19": 19, "LABEL_2": 2, "LABEL_20": 20, "LABEL_21": 21, "LABEL_22": 22, "LABEL_23": 23, "LABEL_24": 24, "LABEL_25": 25, "LABEL_26": 26, "LABEL_27": 27, "LABEL_28": 28, "LABEL_29": 29, "LABEL_3": 3, "LABEL_30": 30, "LABEL_31": 31, "LABEL_32": 32, "LABEL_33": 33, "LABEL_34": 34, "LABEL_35": 35, "LABEL_36": 36, "LABEL_37": 37, "LABEL_38": 38, "LABEL_39": 39, "LABEL_4": 4, "LABEL_40": 40, "LABEL_41": 41, "LABEL_42": 42, "LABEL_43": 43, "LABEL_44": 44, "LABEL_45": 45, "LABEL_46": 46, "LABEL_47": 47, "LABEL_48": 48, "LABEL_49": 49, "LABEL_5": 5, "LABEL_50": 50, "LABEL_51": 51, "LABEL_52": 52, "LABEL_53": 53, "LABEL_54": 54, "LABEL_55": 55, "LABEL_56": 56, "LABEL_57": 57, "LABEL_58": 58, "LABEL_59": 59, "LABEL_6": 6, "LABEL_60": 60, "LABEL_61": 61, "LABEL_62": 62, "LABEL_63": 63, "LABEL_64": 64, "LABEL_65": 65, "LABEL_66": 66, "LABEL_67": 67, "LABEL_68": 68, "LABEL_69": 69, "LABEL_7": 7, "LABEL_70": 70, "LABEL_71": 71, "LABEL_72": 72, "LABEL_73": 73, "LABEL_74": 74, "LABEL_75": 75, "LABEL_76": 76, "LABEL_77": 77, "LABEL_78": 78, "LABEL_79": 79, "LABEL_8": 8, "LABEL_80": 80, "LABEL_81": 81, "LABEL_82": 82, "LABEL_83": 83, "LABEL_84": 84, "LABEL_85": 85, "LABEL_86": 86, "LABEL_87": 87, "LABEL_88": 88, "LABEL_89": 89, "LABEL_9": 9, "LABEL_90": 90 }, "label_noise_prob": 0.5, "loss_alpha": 0.25, "loss_gamma": 2.0, "model_type": "relation_detr", "num_channels": 3, "num_feature_levels": 4, "num_queries": 900, "relation_embed": 16, "torch_dtype": "float32", "transformers_version": "4.46.3", "use_pretrained_backbone": true, "use_timm_backbone": true }