Upload 7 files

README.md

@@ -1,3 +1,57 @@
----
-license: mit
----
+---
+base_model: hustvl/yolos-tiny
+library_name: transformers.js
+---
+https://huggingface.co/hustvl/yolos-tiny with ONNX weights to be compatible with Transformers.js.
+
+## Usage (Transformers.js)
+
+If you haven't already, you can install the [Transformers.js](https://huggingface.co/docs/transformers.js) JavaScript library from [NPM](https://www.npmjs.com/package/@huggingface/transformers) using:
+```bash
+npm i @huggingface/transformers
+```
+
+**Example:** Perform object detection with `Xenova/yolos-tiny`.
+
+```js
+import { pipeline } from "@huggingface/transformers";
+const detector = await pipeline("object-detection", "Xenova/yolos-tiny");
+const image = "https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/cats.jpg";
+const output = await detector(image, { threshold: 0.9 });
+console.log(output);
+```
+
+<details>
+
+<summary>Example output</summary>
+
+```
+[
+  {
+    score: 0.9921281933784485,
+    label: "remote",
+    box: { xmin: 32, ymin: 78, xmax: 185, ymax: 117 },
+  },
+  {
+    score: 0.9884883165359497,
+    label: "remote",
+    box: { xmin: 324, ymin: 82, xmax: 376, ymax: 191 },
+  },
+  {
+    score: 0.9197800159454346,
+    label: "cat",
+    box: { xmin: 5, ymin: 56, xmax: 321, ymax: 469 },
+  },
+  {
+    score: 0.9300552606582642,
+    label: "cat",
+    box: { xmin: 332, ymin: 25, xmax: 638, ymax: 369 },
+  },
+]
+```
+  
+</details>
+
+---
+
+Note: Having a separate repo for ONNX weights is intended to be a temporary solution until WebML gains more traction. If you would like to make your models web-ready, we recommend converting to ONNX using [🤗 Optimum](https://huggingface.co/docs/optimum/index) and structuring your repo like this one (with ONNX weights located in a subfolder named `onnx`).

config.json

@@ -0,0 +1,51 @@
+{
+  "_name_or_path": "CristianR8/Cacao-detection",
+  "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": "fermentado",
+    "1": "insufi_fermen",
+    "2": "pizarroso",
+    "3": "hongo",
+    "4": "insecto",
+    "5": "germinado",
+    "6": "violeta"
+  },
+  "image_size": [
+    800,
+    1333
+  ],
+  "initializer_range": 0.02,
+  "intermediate_size": 768,
+  "label2id": {
+    "fermentado": 0,
+    "insufi_fermen": 1,
+    "pizarroso": 2,
+    "hongo": 3,
+    "insecto": 4,
+    "germinado": 5,
+    "violeta": 6    
+  },
+  "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,
+  "transformers_version": "4.33.0.dev0",
+  "use_mid_position_embeddings": false
+}

onnx/model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cef0d068b114c3076293b8ba8d985561d6d1dbf1db35455919cd0e2c93b5870
+size 10571795

onnx/model_fp16.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3313a6bc63f30ee27bc2bda63a0b8109d8409fba5706cecbafabd3e8a1d69284
+size 5329879

onnx/model_quantized.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7dc437512a2fb98fbfc5692e58b757599a475961aa8b2187a58d373d4cefd85e
+size 2977584

preprocessor_config.json

@@ -0,0 +1,13 @@
+{
+  "do_normalize": false,
+  "do_pad": false,
+  "do_rescale": true,
+  "do_resize": true,
+  "feature_extractor_type": "ImageFeatureExtractor",
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "width": 608,
+    "height": 608
+  }
+}

quantize_config.json

@@ -0,0 +1,26 @@
+{
+    "per_channel": true,
+    "reduce_range": true,
+    "per_model_config": {
+        "model": {
+            "op_types": [
+                "Split",
+                "Concat",
+                "MaxPool",
+                "Slice",
+                "Resize",
+                "Softmax",
+                "Sigmoid",
+                "Conv",
+                "Reshape",
+                "Mul",
+                "Sub",
+                "Transpose",
+                "MatMul",
+                "Div",
+                "Add"
+            ],
+            "weight_type": "QUInt8"
+        }
+    }
+}