This is a crime detection system AI model that uses YOLOv8 to detect crimes in a video feed. The model is trained on various crime datasets and is able to detect crimes in real-time. The model is able to detect upto 15 different classes of crimes.
To use the model, you need run the jupyter notebooks in following order:
Dataset.ipynb
- This notebook is used to download the dataset and prepare it for training.Training.ipynb
- This notebook is used to train the model on the dataset.Detection.ipynb
- This notebook is used to detect crimes in a video feed.
- Open Images Dataset V7
- Weapons Detection Dataset
- Weapon
- Bomb
- Missile
- Sword
- Handgun
- Axe
- Tank
- Shotgun
- Rifle
- Knife
- Dagger
- Cannon
- Bow and arrow
- Pistol
- Gun
This project is licensed under the MIT License - see the LICENSE file for details.