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