bvargas97 / weapon-detection Goto Github PK

View Code? Open in Web Editor NEW

Pistol, Rifle, and Fire detection using yolov4-tiny in videos as well as images. Training code, dataset, and trained weight file available.

Jupyter Notebook 100.00%

weapon-detection's Introduction

weapon-detection

Handgun, Shotgun and Knife using yolov4-tiny in videos as well as images. Training code, dataset and trained weight file available.

Dataset

  Data is Annotated using Labelme and is available in yolo format with txt files
      ├── data
      │    ├── obj (Train dataset)
      │    │   ├── Knife
      │    │   ├── Handgun
      │    │   └── Shotgun
      │    ├── test (Test dataset)
      │    │   ├── Knife
      │    │   ├── Handgun
      │    │   └── Shotgun
      │    ├── train.txt  (Train label)
      │    └── test.txt   (Test label)

DATASET

Pretrained yolo-tiny weights

Pre-train model

Colab GPU Training

Complete project is trained and evaluated on google colab Notebook

Knife, Handgun and Shotgun Detection

Important files for Training

  Training folder contain important files
      ├── training
      │    ├── obj.data
      │    ├── obj.names
      │    ├── yolov4-tiny.config
      │    └── yolov4-tiny.conv.29

Training and Prediction command

 # train your custom detector! (uncomment %%capture below if you run into memory issues or your Colab is crashing)
 # %%capture
 └── !./darknet detector train training/obj.data training/yolov4-tiny.cfg training/yolov4-tiny.conv.29  -dont_show -map
 
 # run your custom detector with this command (upload an image to your google drive to test, thresh flag sets accuracy that detection must be in order to show it)
 ├── !./darknet detector test training/obj.data training/yolov4-tiny.cfg /mydrive/yolo/yolov4/darknet/backup/yolov4-tiny_4000.weights          /mydrive/yolo/yolov4/darknet/data/test/Knife/7bc500661f0ea2c7.jpg -thresh 0.5
 └── imShow('predictions.jpg')