Hugging Face's logo

LykaAustria
/
nicpras_finetuned_yolo

Object Detection
Transformers
yolo
custom-model
finetuned
Inference Endpoints
Model card Files Community
LykaAustria commited on
Commit
c2cf3ef
·
verified ·
1 Parent(s): 6338b4c

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +64 -64
config.yaml CHANGED
@@ -1,64 +1,64 @@
1
- model:
2
- type: YOLOv8
3
- framework: PyTorch
4
- weights: C:\Users\LykaA\Downloads\Lyka sgtech\LEARN\Image Annotation\CVAT\finetune_yolov8\finetuning_yolo\code\runs\detect\train2\weights\best.pt
5
- input_size: [640, 640]
6
- inference:
7
- confidence_threshold: 0.25
8
- iou_threshold: 0.45
9
- max_detections: 300
10
- classes:
11
- - 0: Person
12
- - 1: Scooter
13
- - 2: Wheelchair
14
- - 3: Phone
15
- - 4: Cup
16
- - 5: Book
17
- - 6: Cloth
18
- - 7: Shoes
19
- - 8: Bicycle
20
- - 9: Skating Shoes
21
- - 10: Toy
22
- - 11: Headphone
23
- - 12: Food
24
- - 13: Bottle
25
- - 14: Ball
26
- - 15: Map
27
- - 16: Ice cream
28
- - 17: Camera
29
- - 18: Shovel
30
- - 19: Sandcastle bucket
31
- - 20: Chair
32
- - 21: Trolley
33
- - 22: Swimming float
34
- - 23: Slide
35
- - 24: Water
36
- - 25: Plate
37
- - 26: Tray
38
- - 27: Wallet
39
- - 28: Ipad
40
- - 29: Luggage
41
- - 30: Car
42
- - 31: Package
43
- - 32: Microphone
44
- - 33: Cleaning trolley
45
- - 34: Stroller
46
- - 35: Clothes
47
- - 36: Flyers
48
- - 37: Water can
49
- - 38: Compost
50
- - 39: Pot
51
- - 40: Plant
52
- - 41: Pitcher
53
- - 42: Bin
54
- - 43: Bucket
55
- - 44: Aquarium
56
- - 45: Fishing net
57
- - 46: Sandcastle
58
- - 47: Newspaper
59
- - 48: Shell
60
- - 49: Sunscreen
61
- - 50: Water gun
62
- - 51: Bookshelf
63
- - 52: Yarn
64
- - 53: Baby
 
1
+ model:
2
+ type: YOLOv8
3
+ framework: PyTorch
4
+ weights: ./best.pt
5
+ input_size: [640, 640]
6
+ inference:
7
+ confidence_threshold: 0.25
8
+ iou_threshold: 0.45
9
+ max_detections: 300
10
+ classes:
11
+ - 0: Person
12
+ - 1: Scooter
13
+ - 2: Wheelchair
14
+ - 3: Phone
15
+ - 4: Cup
16
+ - 5: Book
17
+ - 6: Cloth
18
+ - 7: Shoes
19
+ - 8: Bicycle
20
+ - 9: Skating Shoes
21
+ - 10: Toy
22
+ - 11: Headphone
23
+ - 12: Food
24
+ - 13: Bottle
25
+ - 14: Ball
26
+ - 15: Map
27
+ - 16: Ice cream
28
+ - 17: Camera
29
+ - 18: Shovel
30
+ - 19: Sandcastle bucket
31
+ - 20: Chair
32
+ - 21: Trolley
33
+ - 22: Swimming float
34
+ - 23: Slide
35
+ - 24: Water
36
+ - 25: Plate
37
+ - 26: Tray
38
+ - 27: Wallet
39
+ - 28: Ipad
40
+ - 29: Luggage
41
+ - 30: Car
42
+ - 31: Package
43
+ - 32: Microphone
44
+ - 33: Cleaning trolley
45
+ - 34: Stroller
46
+ - 35: Clothes
47
+ - 36: Flyers
48
+ - 37: Water can
49
+ - 38: Compost
50
+ - 39: Pot
51
+ - 40: Plant
52
+ - 41: Pitcher
53
+ - 42: Bin
54
+ - 43: Bucket
55
+ - 44: Aquarium
56
+ - 45: Fishing net
57
+ - 46: Sandcastle
58
+ - 47: Newspaper
59
+ - 48: Shell
60
+ - 49: Sunscreen
61
+ - 50: Water gun
62
+ - 51: Bookshelf
63
+ - 52: Yarn
64
+ - 53: Baby