ArufaDesuwa/Yolo-Pothole-Detection

YOLO Pothole Detection

Model deteksi objek untuk mendeteksi lubang (pothole) dan kerusakan jalan menggunakan YOLO11.

📋 Deskripsi

Model ini merupakan fine-tuned version dari Ultralytics YOLO11 yang dilatih khusus untuk mendeteksi:

• Lubang jalan (pothole)
• Kerusakan jalan (road damage)

Model ini dapat digunakan untuk sistem monitoring kondisi jalan, aplikasi perawatan infrastruktur, atau sistem peringatan dini untuk pengendara.

Dataset

Model ini dilatih menggunakan Pothole Object Detection Dataset dari Roboflow Universe yang terdiri dari:

• 665 gambar dengan anotasi pothole
• Format anotasi: YOLO, COCO JSON, Pascal VOC, dan lainnya
• Dataset telah melalui preprocessing dengan auto-orient
• Tersedia dalam berbagai format untuk berbagai framework deep learning

🚀 Fitur

• Deteksi Real-time: Mampu mendeteksi pothole dalam video atau gambar secara real-time
• Akurasi Tinggi: Dibangun di atas arsitektur YOLO11 yang telah terbukti
• Ringan & Cepat: Optimized untuk deployment di berbagai perangkat
• Multi-bahasa: Mendukung dokumentasi dalam Bahasa Indonesia dan English

📦 Instalasi

Prerequisites

python >= 3.8
pip >= 21.0

Install Dependencies

pip install ultralytics
pip install torch torchvision
pip install opencv-python
pip install huggingface-hub

💻 Cara Penggunaan

1. Load Model dari Hugging Face

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

# Download model
model_path = hf_hub_download(
    repo_id="DityaEn/Yolo-Pothole-Detection",
    filename="best.pt"  # atau nama file model yang sesuai
)

# Load model
model = YOLO(model_path)

2. Deteksi pada Gambar

# Prediksi pada single image
results = model("path/to/image.jpg")

# Tampilkan hasil
for result in results:
    result.show()  # Tampilkan gambar dengan bounding boxes
    
# Atau simpan hasil
results[0].save("output.jpg")

3. Deteksi pada Video

# Prediksi pada video
results = model("path/to/video.mp4")

# Process setiap frame
for result in results:
    result.show()

4. Deteksi Real-time dari Webcam

# Gunakan webcam (0 untuk default camera)
results = model(source=0, show=True)

5. Batch Prediction

# Prediksi multiple images
results = model(["image1.jpg", "image2.jpg", "image3.jpg"])

for i, result in enumerate(results):
    result.save(f"output_{i}.jpg")

📊 Contoh Output

# Mendapatkan deteksi details
results = model("road_image.jpg")

for result in results:
    boxes = result.boxes  # Bounding boxes
    for box in boxes:
        class_id = int(box.cls[0])
        confidence = float(box.conf[0])
        bbox = box.xyxy[0].tolist()  # [x1, y1, x2, y2]
        
        print(f"Class: {model.names[class_id]}")
        print(f"Confidence: {confidence:.2f}")
        print(f"Bounding Box: {bbox}")

🎯 Use Cases

1. Smart City Infrastructure: Monitoring otomatis kondisi jalan
2. Vehicle Safety Systems: Sistem peringatan untuk menghindari pothole
3. Road Maintenance: Identifikasi area yang memerlukan perbaikan
4. Insurance Claims: Dokumentasi kerusakan jalan untuk klaim asuransi
5. Navigation Apps: Integrasi dengan aplikasi navigasi untuk rute alternatif

📈 Model Performance

• Base Model: Ultralytics YOLO11
• Task: Object Detection
• Classes: Pothole, Road Damage
• Input Size: 640x640 (default)
• License: MIT

🤝 Contributing

Kontribusi sangat diterima! Jika Anda ingin berkontribusi:

1. Fork repository ini
2. Buat branch baru (git checkout -b feature/AmazingFeature)
3. Commit perubahan (git commit -m 'Add some AmazingFeature')
4. Push ke branch (git push origin feature/AmazingFeature)
5. Buat Pull Request

📝 License

Project ini dilisensikan di bawah MIT License - lihat file LICENSE untuk detail.

👨‍💻 Author

DityaEn

• Hugging Face: @DityaEn

🙏 Acknowledgments

• Ultralytics untuk YOLO11 base model
• Roboflow untuk Pothole Detection Dataset
• Hugging Face untuk platform model hosting

📞 Contact & Support

Jika Anda memiliki pertanyaan atau masalah:

• Buka issue di Hugging Face Discussions
• Atau hubungi melalui Hugging Face profile

🔗 Links

• Model on Hugging Face
• Dataset on Roboflow
• Ultralytics YOLO11 Documentation
• YOLO11 Paper

---

⭐ Jika project ini bermanfaat, berikan star di Hugging Face!