nicpras_finetuned_yolo / config_cvat.yaml
LykaAustria's picture
Upload config_cvat.yaml
8d4a19f verified
model:
type: YOLOv8
framework: PyTorch
weights: ./best.pt
type: "object-detection"
pipeline_tag: "image-segmentation",
inference:
confidence_threshold: 0.001
classes:
- 0: Person
- 1: Scooter
- 2: Wheelchair
- 3: Phone
- 4: Cup
- 5: Book
- 6: Cloth
- 7: Shoes
- 8: Bicycle
- 9: Skating Shoes
- 10: Toy
- 11: Headphone
- 12: Food
- 13: Bottle
- 14: Ball
- 15: Map
- 16: Ice cream
- 17: Camera
- 18: Shovel
- 19: Sandcastle bucket
- 20: Chair
- 21: Trolley
- 22: Swimming float
- 23: Slide
- 24: Water
- 25: Plate
- 26: Tray
- 27: Wallet
- 28: Ipad
- 29: Luggage
- 30: Car
- 31: Package
- 32: Microphone
- 33: Cleaning trolley
- 34: Stroller
- 35: Clothes
- 36: Flyers
- 37: Water can
- 38: Compost
- 39: Pot
- 40: Plant
- 41: Pitcher
- 42: Bin
- 43: Bucket
- 44: Aquarium
- 45: Fishing net
- 46: Sandcastle
- 47: Newspaper
- 48: Shell
- 49: Sunscreen
- 50: Water gun
- 51: Bookshelf
- 52: Yarn
- 53: Baby