wuhp commited on
Commit
aa77342
·
verified ·
1 Parent(s): 0e776c1

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +54 -0
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # rolo_yolo_fast2
2
+
3
+ ## Model Overview
4
+
5
+ **Architecture:** YOLOv11
6
+ **Training Epochs:** 50
7
+ **Batch Size:** 16
8
+ **Optimizer:** auto
9
+ **Learning Rate:** 0.001
10
+ **Data Augmentation Level:** Basic
11
+
12
+ ## Training Metrics
13
+
14
+ - **[email protected]:** 0.995
15
+
16
+ ## Class IDs
17
+
18
+ | Class ID | Class Name |
19
+ |----------|------------|
20
+ | 0 | Panadol |
21
+ | 1 | Revanin |
22
+
23
+
24
+ ## Datasets Used
25
+
26
+ - Drug Classification
27
+
28
+ ## Class Image Counts
29
+
30
+ | Class Name | Image Count |
31
+ |------------|-------------|
32
+ | Panadol | 116 |
33
+ | Revanin | 122 |
34
+
35
+
36
+ ## Description
37
+
38
+ This model was trained using the YOLOv11 architecture on a custom dataset. The training process involved 50 epochs with a batch size of 16. The optimizer used was **auto** with an initial learning rate of 0.001. Data augmentation was set to the **Basic** level to enhance model robustness.
39
+
40
+ ## Usage
41
+
42
+ To use this model for inference, follow the instructions below:
43
+
44
+ ```python
45
+ from ultralytics import YOLO
46
+
47
+ # Load the trained model
48
+ model = YOLO('best.pt')
49
+
50
+ # Perform inference on an image
51
+ results = model('path_to_image.jpg')
52
+
53
+ # Display results
54
+ results.show()