config.json

{
  "_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
}