This project detects parking spot occupancy using deep learning and computer vision techniques. It includes training and inference scripts for YOLO and CNN models, as well as a Flask web server for visualization and alerting.
- Parking spot detection using YOLOv8 and custom CNN models
- Data collection and augmentation scripts
- Real-time inference and visualization
- Telegram alert integration for wrong parking
- Jupyter notebooks for training and experimentation
main.py- Flask web server and main applicationsend_alert.py- Telegram alert integrationdatacollection.ipynb- Data collection notebookmodel_train/cnn/- CNN training notebooksmodel_train/yolo/- YOLO training and inference notebookscropped_img/- Region of interest imagestemplates/- HTML templates for web UI
- Clone the repository
- Install dependencies:
pip install -r requirements.txt
- Run the Flask server:
python main.py
- Train YOLO model:
Open
model_train/yolo/main.ipynbin Jupyter and run the training cells. - Train CNN model:
Open
model_train/cnn/parking_spot_train_3_classes.ipynbin Jupyter and run the training cells.
See requirements.txt for all required Python packages.
MIT License