Upload YolosForObjectDetection
Browse files- config.json +4 -82
- model.safetensors +1 -1
config.json
CHANGED
@@ -1142,88 +1142,10 @@
|
|
1142 |
"layer_norm_eps": 1e-12,
|
1143 |
"model_type": "yolos",
|
1144 |
"names": [
|
1145 |
-
"
|
1146 |
-
"
|
1147 |
-
"car",
|
1148 |
-
"motorcycle",
|
1149 |
-
"airplane",
|
1150 |
-
"bus",
|
1151 |
-
"train",
|
1152 |
-
"truck",
|
1153 |
-
"boat",
|
1154 |
-
"traffic light",
|
1155 |
-
"fire hydrant",
|
1156 |
-
"stop sign",
|
1157 |
-
"parking meter",
|
1158 |
-
"bench",
|
1159 |
-
"bird",
|
1160 |
-
"cat",
|
1161 |
-
"dog",
|
1162 |
-
"horse",
|
1163 |
-
"sheep",
|
1164 |
-
"cow",
|
1165 |
-
"elephant",
|
1166 |
-
"bear",
|
1167 |
-
"zebra",
|
1168 |
-
"giraffe",
|
1169 |
-
"backpack",
|
1170 |
-
"umbrella",
|
1171 |
-
"handbag",
|
1172 |
-
"tie",
|
1173 |
-
"suitcase",
|
1174 |
-
"frisbee",
|
1175 |
-
"skis",
|
1176 |
-
"snowboard",
|
1177 |
-
"sports ball",
|
1178 |
-
"kite",
|
1179 |
-
"baseball bat",
|
1180 |
-
"baseball glove",
|
1181 |
-
"skateboard",
|
1182 |
-
"surfboard",
|
1183 |
-
"tennis racket",
|
1184 |
-
"bottle",
|
1185 |
-
"wine glass",
|
1186 |
-
"cup",
|
1187 |
-
"fork",
|
1188 |
-
"knife",
|
1189 |
-
"spoon",
|
1190 |
-
"bowl",
|
1191 |
-
"banana",
|
1192 |
-
"apple",
|
1193 |
-
"sandwich",
|
1194 |
-
"orange",
|
1195 |
-
"broccoli",
|
1196 |
-
"carrot",
|
1197 |
-
"hot dog",
|
1198 |
-
"pizza",
|
1199 |
-
"donut",
|
1200 |
-
"cake",
|
1201 |
-
"chair",
|
1202 |
-
"couch",
|
1203 |
-
"potted plant",
|
1204 |
-
"bed",
|
1205 |
-
"dining table",
|
1206 |
-
"toilet",
|
1207 |
-
"tv",
|
1208 |
-
"laptop",
|
1209 |
-
"mouse",
|
1210 |
-
"remote",
|
1211 |
-
"keyboard",
|
1212 |
-
"cell phone",
|
1213 |
-
"microwave",
|
1214 |
-
"oven",
|
1215 |
-
"toaster",
|
1216 |
-
"sink",
|
1217 |
-
"refrigerator",
|
1218 |
-
"book",
|
1219 |
-
"clock",
|
1220 |
-
"vase",
|
1221 |
-
"scissors",
|
1222 |
-
"teddy bear",
|
1223 |
-
"hair drier",
|
1224 |
-
"toothbrush"
|
1225 |
],
|
1226 |
-
"nc":
|
1227 |
"nms_threshold": 0.1,
|
1228 |
"num_attention_heads": 12,
|
1229 |
"num_channels": 3,
|
@@ -1235,6 +1157,6 @@
|
|
1235 |
"torch_dtype": "float32",
|
1236 |
"transformers_version": "4.36.2",
|
1237 |
"use_mid_position_embeddings": true,
|
1238 |
-
"weights": "
|
1239 |
"width_multiple": 1.0
|
1240 |
}
|
|
|
1142 |
"layer_norm_eps": 1e-12,
|
1143 |
"model_type": "yolos",
|
1144 |
"names": [
|
1145 |
+
"0",
|
1146 |
+
"1"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1147 |
],
|
1148 |
+
"nc": 2,
|
1149 |
"nms_threshold": 0.1,
|
1150 |
"num_attention_heads": 12,
|
1151 |
"num_channels": 3,
|
|
|
1157 |
"torch_dtype": "float32",
|
1158 |
"transformers_version": "4.36.2",
|
1159 |
"use_mid_position_embeddings": true,
|
1160 |
+
"weights": "10.pt",
|
1161 |
"width_multiple": 1.0
|
1162 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 419816412
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0bd2422d66175c9c17f1dbf038770dbc3458d98c72e92c7446945b2f351be537
|
3 |
size 419816412
|