wuhp commited on
Commit
72ad2b7
·
verified ·
1 Parent(s): e6b2d0d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +63 -0
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Guns-100-11m
2
+
3
+ ## Model Overview
4
+
5
+ **Architecture:** YOLOv11
6
+ **Training Epochs:** 100
7
+ **Batch Size:** 32
8
+ **Optimizer:** auto
9
+ **Learning Rate:** 0.0005
10
+ **Data Augmentation Level:** Moderate
11
+
12
+ ## Training Metrics
13
+
14
+ - **[email protected]:** 0.85476
15
+
16
+ ## Class IDs
17
+
18
+ | Class ID | Class Name |
19
+ |----------|------------|
20
+ | 0 | Gun |
21
+ | 1 | Knife |
22
+
23
+
24
+ ## Datasets Used
25
+
26
+ - gun-detection-xwosb_v8
27
+ - gun-sqvhj_v1
28
+ - people_with_guns_2-pvcnk_v9
29
+ - police1_v1
30
+ - soldier-wclp7_v1
31
+ - test-e46au_v1
32
+ - weapon-rl8c4_v3
33
+ - weapondetection-fdzlo_v4
34
+ - weapons-detection-x9fnq_v3
35
+ - wepon_detection_v1
36
+
37
+ ## Class Image Counts
38
+
39
+ | Class Name | Image Count |
40
+ |------------|-------------|
41
+ | Gun | 44219 |
42
+ | Knife | 4038 |
43
+
44
+
45
+ ## Description
46
+
47
+ This model was trained using the YOLOv11 architecture on a custom dataset. The training process involved 100 epochs with a batch size of 32. The optimizer used was **auto** with an initial learning rate of 0.0005. Data augmentation was set to the **Moderate** level to enhance model robustness.
48
+
49
+ ## Usage
50
+
51
+ To use this model for inference, follow the instructions below:
52
+
53
+ ```python
54
+ from ultralytics import YOLO
55
+
56
+ # Load the trained model
57
+ model = YOLO('best.pt')
58
+
59
+ # Perform inference on an image
60
+ results = model('path_to_image.jpg')
61
+
62
+ # Display results
63
+ results.show()