Upload YolosForObjectDetection

config.json

@@ -1,130 +1,130 @@
 {
-   "_name_or_path":"valentinafeve/yolos-fashionpedia",
-   "architectures":[
-      "YolosForObjectDetection"
-   ],
-   "attention_probs_dropout_prob":0.0,
-   "auxiliary_loss":false,
-   "bbox_cost":5,
-   "bbox_loss_coefficient":5,
-   "class_cost":1,
-   "eos_coefficient":0.1,
-   "giou_cost":2,
-   "giou_loss_coefficient":2,
-   "hidden_act":"gelu",
-   "hidden_dropout_prob":0.0,
-   "hidden_size":384,
-   "id2label":{
-      "0":"shirt, blouse",
-      "1":"top, t-shirt, sweatshirt",
-      "2":"sweater",
-      "3":"cardigan",
-      "4":"jacket",
-      "5":"vest",
-      "6":"pants",
-      "7":"shorts",
-      "8":"skirt",
-      "9":"coat",
-      "10":"dress",
-      "11":"jumpsuit",
-      "12":"cape",
-      "13":"glasses",
-      "14":"hat",
-      "15":"headband, head covering, hair accessory",
-      "16":"tie",
-      "17":"glove",
-      "18":"watch",
-      "19":"belt",
-      "20":"leg warmer",
-      "21":"tights, stockings",
-      "22":"sock",
-      "23":"shoe",
-      "24":"bag, wallet",
-      "25":"scarf",
-      "26":"umbrella",
-      "27":"hood",
-      "28":"collar",
-      "29":"lapel",
-      "30":"epaulette",
-      "31":"sleeve",
-      "32":"pocket",
-      "33":"neckline",
-      "34":"buckle",
-      "35":"zipper",
-      "36":"applique",
-      "37":"bead",
-      "38":"bow",
-      "39":"flower",
-      "40":"fringe",
-      "41":"ribbon",
-      "42":"rivet",
-      "43":"ruffle",
-      "44":"sequin",
-      "45":"tassel"
-   },
-   "image_size":[
-      512,
-      864
-   ],
-   "initializer_range":0.02,
-   "intermediate_size":1536,
-   "label2id":{
-      "shirt, blouse":0,
-      "top, t-shirt, sweatshirt":1,
-      "sweater":2,
-      "cardigan":3,
-      "jacket":4,
-      "vest":5,
-      "pants":6,
-      "shorts":7,
-      "skirt":8,
-      "coat":9,
-      "dress":10,
-      "jumpsuit":11,
-      "cape":12,
-      "glasses":13,
-      "hat":14,
-      "headband, head covering, hair accessory":15,
-      "tie":16,
-      "glove":17,
-      "watch":18,
-      "belt":19,
-      "leg warmer":20,
-      "tights, stockings":21,
-      "sock":22,
-      "shoe":23,
-      "bag, wallet":24,
-      "scarf":25,
-      "umbrella":26,
-      "hood":27,
-      "collar":28,
-      "lapel":29,
-      "epaulette":30,
-      "sleeve":31,
-      "pocket":32,
-      "neckline":33,
-      "buckle":34,
-      "zipper":35,
-      "applique":36,
-      "bead":37,
-      "bow":38,
-      "flower":39,
-      "fringe":40,
-      "ribbon":41,
-      "rivet":42,
-      "ruffle":43,
-      "sequin":44,
-      "tassel":45
-   },
-   "layer_norm_eps":1e-12,
-   "model_type":"yolos",
-   "num_attention_heads":6,
-   "num_channels":3,
-   "num_detection_tokens":100,
-   "num_hidden_layers":12,
-   "patch_size":16,
-   "qkv_bias":true,
-   "torch_dtype":"float32",
-   "transformers_version":"4.23.1",
-   "use_mid_position_embeddings":true
-}
+  "_name_or_path": "hustvl/yolos-small",
+  "architectures": [
+    "YolosForObjectDetection"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "auxiliary_loss": false,
+  "bbox_cost": 5,
+  "bbox_loss_coefficient": 5,
+  "class_cost": 1,
+  "eos_coefficient": 0.1,
+  "giou_cost": 2,
+  "giou_loss_coefficient": 2,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 384,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3",
+    "4": "LABEL_4",
+    "5": "LABEL_5",
+    "6": "LABEL_6",
+    "7": "LABEL_7",
+    "8": "LABEL_8",
+    "9": "LABEL_9",
+    "10": "LABEL_10",
+    "11": "LABEL_11",
+    "12": "LABEL_12",
+    "13": "LABEL_13",
+    "14": "LABEL_14",
+    "15": "LABEL_15",
+    "16": "LABEL_16",
+    "17": "LABEL_17",
+    "18": "LABEL_18",
+    "19": "LABEL_19",
+    "20": "LABEL_20",
+    "21": "LABEL_21",
+    "22": "LABEL_22",
+    "23": "LABEL_23",
+    "24": "LABEL_24",
+    "25": "LABEL_25",
+    "26": "LABEL_26",
+    "27": "LABEL_27",
+    "28": "LABEL_28",
+    "29": "LABEL_29",
+    "30": "LABEL_30",
+    "31": "LABEL_31",
+    "32": "LABEL_32",
+    "33": "LABEL_33",
+    "34": "LABEL_34",
+    "35": "LABEL_35",
+    "36": "LABEL_36",
+    "37": "LABEL_37",
+    "38": "LABEL_38",
+    "39": "LABEL_39",
+    "40": "LABEL_40",
+    "41": "LABEL_41",
+    "42": "LABEL_42",
+    "43": "LABEL_43",
+    "44": "LABEL_44",
+    "45": "LABEL_45"
+  },
+  "image_size": [
+    512,
+    864
+  ],
+  "initializer_range": 0.02,
+  "intermediate_size": 1536,
+  "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_5": 5,
+    "LABEL_6": 6,
+    "LABEL_7": 7,
+    "LABEL_8": 8,
+    "LABEL_9": 9
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "yolos",
+  "num_attention_heads": 6,
+  "num_channels": 3,
+  "num_detection_tokens": 100,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.23.1",
+  "use_mid_position_embeddings": true
+}

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6b20b44b21acfe8e918634da01a21eb8e4e326995cd0b6869e423ea7dbdbd334
+oid sha256:8c369dd8d593421b25889a85f1689e29c356522abdb7be71c99e7ef939473aa8
 size 122738781