cakra84 commited on
Commit
ae1ecc3
·
verified ·
1 Parent(s): 1285781

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +62 -3
README.md CHANGED
@@ -1,3 +1,62 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - computer-vision
5
+ - object-detection
6
+ - pytorch
7
+ - yolov7
8
+ - road-damage-detection
9
+ - autonomous-driving
10
+ datasets:
11
+ - custom-pothole-dataset
12
+ widget:
13
+ - src: "https://placehold.co/640x480/2d3748/ffffff?text=Road+Damage+Example"
14
+ example_title: "Road with Pothole"
15
+ ---
16
+
17
+ # YOLOv7-tiny for Road Damage Detection (Including Water-Filled Potholes)
18
+
19
+ This is a `YOLOv7-tiny` model trained to detect various types of road damage, with a special focus on identifying potholes that are filled with water. The model is based on the final project research by **Benito Yvan Deva Putra Arung Dirgantara** from the Politeknik Elektronika Negeri Surabaya (PENS).
20
+
21
+ The project systematically analyzes the detection of road defects using a Convolutional Neural Network (CNN) approach, evaluates optimal training configurations through extensive hyperparameter tuning, and assesses deployment performance on edge devices like the NVIDIA Jetson AGX Orin.
22
+
23
+ ## Model Description
24
+
25
+ - **Architecture:** YOLOv7-tiny
26
+ - **Framework:** PyTorch
27
+ - **Objective:** To detect and classify four types of road damage for real-time road maintenance and safety applications.
28
+ - **Classes:**
29
+ - `L00`: Dry Pothole
30
+ - `L01`: Wet Pothole (water-filled)
31
+ - `R02`: Longitudinal Crack
32
+ - `R03`: Alligator Crack
33
+
34
+ ## Intended Use
35
+
36
+ This model is intended for real-time road damage detection from a moving vehicle. It has been optimized and tested for deployment on edge AI hardware, such as the NVIDIA Jetson series, making it suitable for integration into road survey vehicles or autonomous driving systems. The optimal detection speed was found to be between **10-15 km/h**.
37
+
38
+ ## How to Use
39
+
40
+ You can use this model with the `ultralytics` library or a standard PyTorch implementation for YOLOv7.
41
+
42
+ ```python
43
+ import torch
44
+
45
+ # Load the model (replace with the actual path to your .pt file)
46
+ # For example, you can load it from the Hugging Face Hub:
47
+ # model = torch.hub.load('path/to/yolov7', 'custom', path='your_model.pt', source='local')
48
+ # Or if using a compatible library:
49
+ # from ultralytics import YOLO
50
+ # model = YOLO('your_model.pt')
51
+
52
+ # Path to your image
53
+ image_path = 'path/to/your/road_image.jpg'
54
+
55
+ # Perform inference
56
+ results = model(image_path)
57
+
58
+ # Print results
59
+ results.print()
60
+
61
+ # Show results
62
+ results.show()