kesimeg's picture
Create README.md
184f201 verified
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?

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()
kesimeg/yolov8n-clothing-detection