File size: 1,545 Bytes
8842569 c2b7bf4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
---
license: mit
language: en
tags:
- pytorch
- image-classification
- computer-vision
- pothole-detection
---
# 🧠 Pothole Detection Model (ResNet18)
This repository contains a fine-tuned **ResNet18** model for classifying whether a road image contains a **pothole** or not.
The model was developed as part of the **Zindi MIIA Pothole Image Classification Challenge**, focused on improving road safety in South Africa.
## 📚 Project Overview
- **Task:** Binary image classification (`pothole` / `no_pothole`)
- **Framework:** PyTorch + torchvision
- **Model:** ResNet18 (pretrained on ImageNet, fine-tuned)
- **Evaluation Metric:** AUC (Area Under the Curve)
- **Dataset Source:** [rupesh002/Patholes_Dataset](https://huggingface.co/datasets/rupesh002/Patholes_Dataset)
---
## 🔗 Full Project Code
The complete project (training, inference, and notebook) is available on GitHub:
👉 [MIIA Pothole Image Classification on GitHub](https://github.com/Rupeshbhardwaj002/MIIA_Pothole_Image_Classification_SouthAfrica)
It includes:
- Data preprocessing
- Model training pipeline (`train.py`)
- Prediction and submission script (`predict.py`)
- Notebook used for Zindi competition submission
---
## ⚙️ How to Load This Model
```python
from huggingface_hub import hf_hub_download
import joblib
model_path = hf_hub_download(
repo_id="rupesh002/pothole_detection_model",
filename="model.pkl",
repo_type="model"
)
model = joblib.load(model_path)
---
🏷️ Author
Rupesh Bhardwaj
Machine Learning Enthusiast |