Upload 4 files

README.md

@@ -1,3 +1,59 @@
 ---
-license: unknown
+library_name: transformers
+pipeline_tag: image-segmentation
+tags:
+  - image-segmentation
+  - mask-generation
+  - transformers.js
+  - vision
+  - background-removal
+  - portrait-matting
+license: apache-2.0
+language:
+  - en
 ---
+# U-2-Netp
+
+## Model Description
+U-2-Netp is a lightweight version of the U2Net model designed for efficient and effective image segmentation tasks, especially for generating masks. It retains the core architectural design of U2Net while being optimized for faster inference times and reduced memory usage.
+
+## Usage
+Perform mask generation with `BritishWerewolf/U-2-Netp`.
+
+### Example
+```javascript
+import { AutoModel, AutoProcessor, RawImage } from '@huggingface/transformers';
+
+const img_url = 'https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png';
+const image = await RawImage.read(img_url);
+
+const processor = await AutoProcessor.from_pretrained('BritishWerewolf/U-2-Netp');
+const processed = await processor(image);
+
+const model = await AutoModel.from_pretrained('BritishWerewolf/U-2-Netp', {
+    dtype: 'fp32',
+});
+
+const output = await model({ input: processed.pixel_values });
+// {
+//   mask: Tensor {
+//     dims: [ 1, 320, 320 ],
+//     type: 'uint8',
+//     data: Uint8Array(102400) [ ... ],
+//     size: 102400
+//   }
+// }
+```
+
+## Model Architecture
+The U-2-Netp model is based on a simplified version of the original U2Net architecture, designed to be more lightweight while still achieving high performance in segmentation tasks. The model consists of several stages with down-sampling and up-sampling paths, using Residual U-blocks (RSU) for enhanced feature representation.
+
+### Inference
+To use the model for inference, you can follow the example provided above. The `AutoProcessor` and `AutoModel` classes from the `transformers` library make it easy to load the model and processor.
+
+## Credits
+* [`rembg`](https://github.com/danielgatis/rembg) for the ONNX model.
+* The authors of the original U-2-Net model can be credited at https://github.com/xuebinqin/U-2-Net.
+
+## Licence
+This model is licensed under the Apache License 2.0 to match the original U-2-Net model.

config.json

@@ -0,0 +1,23 @@
+{
+  "_name_or_path": "u2netp",
+  "model_type": "u2net",
+  "architectures": [
+    "U2NetModel"
+  ],
+  "transformers.js_config": {
+    "dtype": "fp32"
+  },
+  "input_name": "input.1",
+  "input_shape": [1, 3, 320, 320],
+  "output_composite": "1959",
+  "output_names": [
+    "1959",
+    "1960",
+    "1961",
+    "1962",
+    "1963",
+    "1964",
+    "1965"
+  ],
+  "output_shape": [1, 320, 320]
+}

gitattributes.txt

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

preprocessor_config.json

@@ -0,0 +1,27 @@
+{
+  "processor_class": "U2NetProcessor",
+  "image_processor_type": "U2NetImageProcessor",
+  "do_convert_rgb": true,
+  "do_normalize": true,
+  "do_pad": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "keep_aspect_ratio": true,
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "pad_size": {
+    "width": 320,
+    "height": 320
+  },
+  "size": {
+    "longest_edge": 320
+  }
+}