yolos-fashionpedia / config.json
valentinafeve's picture
Update config.json
4776553
raw
history blame
2.79 kB
{
"_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
}