/weapon_detection_model

Transfer learning a Faster-RCNN model to predict 4 different classes of weapon using detectron2 framework

Primary LanguageJupyter Notebook

weapon_detection_model

Transfer learning a Faster-RCNN model to predict 4 different classes of weapon using detectron2 framework

classes

Handgun
Knife
Rifle
Sword

(In the notebook you may see another extra class weapons, which is generated by roboflow on preprocessing the dataset you can ignore)

notebook workflow

  1. install detectron2
  2. use pre-trained models mask-rcnn, frcnn on a random image
  3. train frcnn on custom dataset
  4. test and evaluate

dataset

dataset

trained model

model