Initial commit

.gitattributes

@@ -0,0 +1,34 @@
+*.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
+*.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

README.md

@@ -0,0 +1,76 @@
+---
+license_name: server-side-public-license
+license_link: https://www.mongodb.com/licensing/server-side-public-license
+tags:
+- fashion
+- cloth-retrieval
+- e-commerce
+- segmentation
+datasets:
+- rizavelioglu/fashionfail
+- detection-datasets/fashionpedia
+pipeline_tag: object-detection
+---
+
+## Models
+
+The models proposed in the paper ["FashionFail: Addressing Failure Cases in Fashion Object Detection and Segmentation"](https://arxiv.org/abs/2404.08582)
+
+## Usage
+
+```python
+from torchvision.io import read_image
+from torchvision.models.detection import MaskRCNN_ResNet50_FPN_Weights
+from huggingface_hub import hf_hub_download
+
+path_onnx = hf_hub_download(
+    repo_id="rizavelioglu/segmentation",
+    filename="facere_base.onnx",  # or "facere_plus.onnx"
+)
+
+# Load pre-trained model transformations.
+weights = MaskRCNN_ResNet50_FPN_Weights.DEFAULT
+transforms = weights.transforms()
+
+# Load image and apply original transformation to the image.
+img = read_image(image)
+img_transformed = transforms(img)
+
+# Create an inference session.
+ort_session = onnxruntime.InferenceSession(
+    path_onnx, providers=["CUDAExecutionProvider", "CPUExecutionProvider"]
+)
+
+# Run inference on the input.
+ort_inputs = {
+    ort_session.get_inputs()[0].name: img_transformed.unsqueeze(dim=0).numpy()
+}
+ort_outs = ort_session.run(None, ort_inputs)
+
+# Parse the model output.
+boxes, labels, scores, masks = ort_outs
+```
+
+> Check out the demo code on [HuggingFace Spaces][ff-hf_spaces] for visualizing the output.
+
+> Also, check out [FashionFail's GitHub repository](https://github.com/rizavelioglu/fashionfail) to get more information on
+> training, inference, and evaluation.
+
+### License
+TL;DR: Not available for commercial use, unless the FULL source code is shared! \
+This project is intended solely for academic research. No commercial benefits are derived from it.
+Models are licensed under [Server Side Public License (SSPL)](https://www.mongodb.com/legal/licensing/server-side-public-license)
+
+### Citation
+If you find this repository useful in your research, please consider giving a star ⭐ and a citation:
+```
+@inproceedings{velioglu2024fashionfail,
+  author    = {Velioglu, Riza and Chan, Robin and Hammer, Barbara},
+  title     = {FashionFail: Addressing Failure Cases in Fashion Object Detection and Segmentation},
+  journal   = {IJCNN},
+  eprint    = {2404.08582},
+  year      = {2024},
+}
+```
+
+[ff-hf_spaces]: https://huggingface.co/spaces/rizavelioglu/fashionfail

facere_base.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01f6d63a33ab14ab317c094e1c6948f2f4cab386b2a2f8840f03779916709363
+size 184743537

facere_plus.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:962749b2343e6b6fe498df2328683be5c5db315590ea179d94cbcf28d7629103
+size 184226861