detect_car4-yolos / config.json
amusktweewt's picture
Update config.json
18f61e6 verified
{
"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": 192,
"id2label": {
"0": "car",
"1": "parked_car",
"2": "truck",
"3": "parked_truck",
"4": "bus",
"5": "parked_bus",
"6": "motorbike",
"7": "parked_motorbike",
"8": "bicycle",
"9": "parked_bicycle",
"10": "pedestrian",
"11": "tunnel_sign",
"12": "light_sign",
"13": "no_light_sign",
"14": "no_entry_sign",
"15": "traffic_light",
"16": "stop_sign",
"17": "stop_sign_road",
"18": "yield_sign",
"19": "yield_sign_road",
"20": "front_arrow_road",
"21": "left_arrow_road",
"22": "right_arrow_road",
"23": "front_left_arrow_road",
"24": "front_right_arrow_road",
"25": "no_left_turn_sign",
"26": "no_right_turn_sign",
"27": "no_u_turn_sign",
"28": "right_aggressive_turn_right_sign",
"29": "left_aggressive_turn_left_sign",
"30": "no_overtaking_sign",
"31": "no_speed_limit_sign",
"32": "pedestrian_crossing_sign",
"33": "pedestrian_crossing",
"34": "sticks",
"35": "trash_cans",
"36": "km_point_sign",
"37": "two_way_sign",
"38": "side_road_warning_sign",
"39": "cross_road_warning_sign",
"40": "roundabout_sign",
"41": "roundabout_warning_sign",
"42": "bus_stop_sign",
"43": "ban_overtake_sign",
"44": "speed_limit_sign",
"45": "speed_limit_sign_road",
"46": "stop_road_lane",
"47": "van",
"48": "parked_van",
"49": "red_traffic_light",
"50": "yellow_traffic_light",
"51": "green_traffic_light",
"52": "cone"
},
"image_size": [
414,
736
],
"initializer_range": 0.02,
"intermediate_size": 768,
"label2id": {
"car": 0,
"parked_car": 1,
"truck": 2,
"parked_truck": 3,
"bus": 4,
"parked_bus": 5,
"motorbike": 6,
"parked_motorbike": 7,
"bicycle": 8,
"parked_bicycle": 9,
"pedestrian": 10,
"tunnel_sign": 11,
"light_sign": 12,
"no_light_sign": 13,
"no_entry_sign": 14,
"traffic_light": 15,
"stop_sign": 16,
"stop_sign_road": 17,
"yield_sign": 18,
"yield_sign_road": 19,
"front_arrow_road": 20,
"left_arrow_road": 21,
"right_arrow_road": 22,
"front_left_arrow_road": 23,
"front_right_arrow_road": 24,
"no_left_turn_sign": 25,
"no_right_turn_sign": 26,
"no_u_turn_sign": 27,
"right_aggressive_turn_right_sign": 28,
"left_aggressive_turn_left_sign": 29,
"no_overtaking_sign": 30,
"no_speed_limit_sign": 31,
"pedestrian_crossing_sign": 32,
"pedestrian_crossing": 33,
"sticks": 34,
"trash_cans": 35,
"km_point_sign": 36,
"two_way_sign": 37,
"side_road_warning_sign": 38,
"cross_road_warning_sign": 39,
"roundabout_sign": 40,
"roundabout_warning_sign": 41,
"bus_stop_sign": 42,
"ban_overtake_sign": 43,
"speed_limit_sign": 44,
"speed_limit_sign_road": 45,
"stop_road_lane": 46,
"van": 47,
"parked_van": 48,
"red_traffic_light": 49,
"yellow_traffic_light": 50,
"green_traffic_light": 51,
"cone": 52
},
"layer_norm_eps": 1e-12,
"model_type": "yolos",
"num_attention_heads": 3,
"num_channels": 3,
"num_detection_tokens": 100,
"num_hidden_layers": 12,
"patch_size": 16,
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.19.0.dev0",
"use_mid_position_embeddings": false
}