eeshawn11 commited on
Commit
91ef390
1 Parent(s): d6e91ed

add ultralytics model card

Browse files
Files changed (1) hide show
  1. README.md +68 -7
README.md CHANGED
@@ -1,10 +1,71 @@
 
1
  ---
2
- license: gpl-3.0
3
- language:
4
- - en
5
- pipeline_tag: object-detection
6
  tags:
7
- - naruto
8
- - hand seals
9
  - yolov8
10
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
  ---
 
 
 
 
3
  tags:
4
+ - ultralyticsplus
 
5
  - yolov8
6
+ - ultralytics
7
+ - yolo
8
+ - vision
9
+ - object-detection
10
+ - pytorch
11
+
12
+ library_name: ultralytics
13
+ library_version: 8.0.43
14
+ inference: false
15
+
16
+ model-index:
17
+ - name: eeshawn11/naruto_hand_seal_detection
18
+ results:
19
+ - task:
20
+ type: object-detection
21
+
22
+ metrics:
23
+ - type: precision # since mAP@0.5 is not available on hf.co/metrics
24
+ value: 0.995 # min: 0.0 - max: 1.0
25
+ name: mAP@0.5(box)
26
+ ---
27
+
28
+ <div align="center">
29
+ <img width="640" alt="eeshawn11/naruto_hand_seal_detection" src="https://huggingface.co/eeshawn11/naruto_hand_seal_detection/resolve/main/thumbnail.jpg">
30
+ </div>
31
+
32
+ ### Supported Labels
33
+
34
+ ```
35
+ ['bird', 'boar', 'dog', 'dragon', 'hare', 'horse', 'monkey', 'ox', 'ram', 'rat', 'snake', 'tiger']
36
+ ```
37
+
38
+ ### How to use
39
+
40
+ - Install [ultralyticsplus](https://github.com/fcakyon/ultralyticsplus):
41
+
42
+ ```bash
43
+ pip install ultralyticsplus==0.0.28 ultralytics==8.0.43
44
+ ```
45
+
46
+ - Load model and perform prediction:
47
+
48
+ ```python
49
+ from ultralyticsplus import YOLO, render_result
50
+
51
+ # load model
52
+ model = YOLO('eeshawn11/naruto_hand_seal_detection')
53
+
54
+ # set model parameters
55
+ model.overrides['conf'] = 0.25 # NMS confidence threshold
56
+ model.overrides['iou'] = 0.45 # NMS IoU threshold
57
+ model.overrides['agnostic_nms'] = False # NMS class-agnostic
58
+ model.overrides['max_det'] = 1000 # maximum number of detections per image
59
+
60
+ # set image
61
+ image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
62
+
63
+ # perform inference
64
+ results = model.predict(image)
65
+
66
+ # observe results
67
+ print(results[0].boxes)
68
+ render = render_result(model=model, image=image, result=results[0])
69
+ render.show()
70
+ ```
71
+