alreaper commited on
Commit
d02910f
·
1 Parent(s): 32f448a

Add trained YOLO model checkpoints

Browse files
Files changed (3) hide show
  1. README.md +38 -0
  2. best.pt +3 -0
  3. last.pt +3 -0
README.md CHANGED
@@ -1,3 +1,41 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ tags:
4
+ - object-detection
5
+ - yolo
6
+ - water-meter
7
+ - computer-vision
8
  ---
9
+
10
+ # Rwanda Water Meter Reading Model
11
+
12
+ This repository contains a YOLO model trained to detect:
13
+
14
+ - water meter
15
+ - reading window
16
+ - digits `0` through `9`
17
+ - unreadable digits (`unknown`)
18
+
19
+ ## Files
20
+
21
+ - `best.pt`: best checkpoint from training
22
+ - `last.pt`: most recent checkpoint
23
+
24
+ ## Training source
25
+
26
+ The model was trained on the prepared dataset from the Rwanda water meter
27
+ reading project.
28
+
29
+ ## Suggested usage
30
+
31
+ ```python
32
+ from ultralytics import YOLO
33
+
34
+ model = YOLO("best.pt")
35
+ results = model.predict("image.jpg", imgsz=640)
36
+ ```
37
+
38
+ ## Notes
39
+
40
+ - `best.pt` is the recommended checkpoint for inference
41
+ - `last.pt` can be used to continue training
best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7215bf64c237d7b376c4f0ac85141ace5b75acdbf5f8be805e2b8ac573269167
3
+ size 6219050
last.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ddc392cff38efbf83d8352f8ecfbc7461a37eca38553fef4db6564bc157aad3
3
+ size 6219050