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
|