Update config.json
Browse files- config.json +11 -4
config.json
CHANGED
@@ -1142,10 +1142,17 @@
|
|
1142 |
"layer_norm_eps": 1e-12,
|
1143 |
"model_type": "yolos",
|
1144 |
"names": [
|
1145 |
-
"
|
1146 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1147 |
],
|
1148 |
-
"nc":
|
1149 |
"nms_threshold": 0.1,
|
1150 |
"num_attention_heads": 12,
|
1151 |
"num_channels": 3,
|
@@ -1157,6 +1164,6 @@
|
|
1157 |
"torch_dtype": "float32",
|
1158 |
"transformers_version": "4.36.2",
|
1159 |
"use_mid_position_embeddings": true,
|
1160 |
-
"weights": "
|
1161 |
"width_multiple": 1.0
|
1162 |
}
|
|
|
1142 |
"layer_norm_eps": 1e-12,
|
1143 |
"model_type": "yolos",
|
1144 |
"names": [
|
1145 |
+
"person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck", "boat", "traffic light",
|
1146 |
+
"fire hydrant", "stop sign", "parking meter", "bench", "bird", "cat", "dog", "horse", "sheep", "cow",
|
1147 |
+
"elephant", "bear", "zebra", "giraffe", "backpack", "umbrella", "handbag", "tie", "suitcase", "frisbee",
|
1148 |
+
"skis", "snowboard", "sports ball", "kite", "baseball bat", "baseball glove", "skateboard", "surfboard",
|
1149 |
+
"tennis racket", "bottle", "wine glass", "cup", "fork", "knife", "spoon", "bowl", "banana", "apple",
|
1150 |
+
"sandwich", "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch",
|
1151 |
+
"potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse", "remote", "keyboard", "cell phone",
|
1152 |
+
"microwave", "oven", "toaster", "sink", "refrigerator", "book", "clock", "vase", "scissors", "teddy bear",
|
1153 |
+
"hair drier", "toothbrush"
|
1154 |
],
|
1155 |
+
"nc": 80,
|
1156 |
"nms_threshold": 0.1,
|
1157 |
"num_attention_heads": 12,
|
1158 |
"num_channels": 3,
|
|
|
1164 |
"torch_dtype": "float32",
|
1165 |
"transformers_version": "4.36.2",
|
1166 |
"use_mid_position_embeddings": true,
|
1167 |
+
"weights": "yolov7.pt",
|
1168 |
"width_multiple": 1.0
|
1169 |
}
|