File size: 1,170 Bytes
184f201 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
---
datasets:
- detection-datasets/fashionpedia_4_categories
language:
- en
base_model:
- Ultralytics/YOLOv8
pipeline_tag: object-detection
---
This model is a finetuned version of YOLO-v8n on clothing detection.
### Object categories
- Clothing
- Shoes
- Bags
- Accessories
You can get more information (and code 🎉) on how to train or use the model on my [github].
[github]: https://github.com/kesimeg/YOLO-Clothing-Detection
# How to use the model?
- Install [ultralyticsplus](https://github.com/fcakyon/ultralyticsplus):
```bash
pip install ultralyticsplus==0.0.23 ultralytics==8.0.21
```
- Load model and perform prediction:
```python
from ultralyticsplus import YOLO, render_result
# load model
model = YOLO('kesimeg/yolov8n-clothing-detection')
# set image
image = 'some_image.jpg'
# perform inference
results = model.predict(image)
# observe results
print(results[0].boxes)
render = render_result(model=model, image=image, result=results[0])
render.show()
```
<div align="center">
<img width="640" alt="kesimeg/yolov8n-clothing-detection" src="https://huggingface.co/kesimeg/yolov8n-clothing-detection/resolve/main/sample_output.png">
</div> |