jonalvarez01 commited on
Commit
a9cb3fe
·
verified ·
1 Parent(s): 6be789c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +46 -2
README.md CHANGED
@@ -1,6 +1,50 @@
1
  ---
2
  language:
3
- - es
4
  base_model:
5
  - Ultralytics/YOLOv8
6
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  language:
3
+ - en
4
  base_model:
5
  - Ultralytics/YOLOv8
6
+ ---
7
+ # Mobile and Cap Detection Model - YOLOv8
8
+
9
+ This repository contains a YOLOv8 model trained to detect mobile phones and caps in images.
10
+
11
+ ## Model Description
12
+
13
+ The model was trained using YOLOv8n architecture to detect two classes:
14
+ - Mobile phones
15
+ - Caps
16
+
17
+ ### Training Details
18
+ - Base model: YOLOv8n
19
+ - Training epochs: 100
20
+ - Hardware: CUDA-enabled GPU
21
+ - Framework: Ultralytics YOLO
22
+
23
+ ## Usage
24
+
25
+ ```python
26
+ from ultralytics import YOLO
27
+
28
+ # Load the model
29
+ model = YOLO('path_to_model.pt')
30
+
31
+ # Perform detection on an image
32
+ results = model('path_to_image.jpg')
33
+ ```
34
+
35
+ ## Training Code
36
+
37
+ The model was trained using the following script:
38
+
39
+ ```python
40
+ from ultralytics import YOLO
41
+
42
+ # Load a model
43
+ model = YOLO("yolov8n.pt").to('cuda')
44
+ # Train the model
45
+ results = model.train(data="data.yaml", epochs=100)
46
+ ```
47
+
48
+ ## Dataset
49
+
50
+ The model was trained on a custom dataset containing images of mobile phones and caps. The dataset was structured following YOLO format requirements.