sergioska
/

yolo-finetuned-flyers

Object Detection

Inference Endpoints

Model card Files Files and versions Community

sergioska commited on Dec 15, 2023

Commit

5501f0f

•

1 Parent(s): e9b7fcf

Upload YolosForObjectDetection

Files changed (2) hide show

config.json +6 -6
model.safetensors +2 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "hustvl/yolos-tiny",
   "architectures": [
     "YolosForObjectDetection"
   ],
@@ -13,16 +13,16 @@
   "giou_loss_coefficient": 2,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 192,
   "image_size": [
     800,
-    1333
   ],
   "initializer_range": 0.02,
-  "intermediate_size": 768,
   "layer_norm_eps": 1e-12,
   "model_type": "yolos",
-  "num_attention_heads": 3,
   "num_channels": 3,
   "num_detection_tokens": 100,
   "num_hidden_layers": 12,
@@ -30,5 +30,5 @@
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.35.2",
-  "use_mid_position_embeddings": false
 }

 {
+  "_name_or_path": "hustvl/yolos-base",
   "architectures": [
     "YolosForObjectDetection"
   ],
   "giou_loss_coefficient": 2,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
   "image_size": [
     800,
+    1344
   ],
   "initializer_range": 0.02,
+  "intermediate_size": 3072,
   "layer_norm_eps": 1e-12,
   "model_type": "yolos",
+  "num_attention_heads": 12,
   "num_channels": 3,
   "num_detection_tokens": 100,
   "num_hidden_layers": 12,
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.35.2",
+  "use_mid_position_embeddings": true
 }

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:318ed064492d50c3fc946f61291639f9f239256bff2f0c1e01de3ec21998d9fa
-size 25910172

 version https://git-lfs.github.com/spec/v1
+oid sha256:956315fdb0a7d43c787c73b4fa31a39cf6f93ed36e301548cdfea958931ac381
+size 510944236