metadata
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.
How to use the model?
- Install ultralyticsplus:
pip install ultralyticsplus==0.0.23 ultralytics==8.0.21
- Load model and perform prediction:
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()