/Weapon_Detection

A barebone model built for detecting weapons in an image. It was built using YOLOv2 (You Only Look Once Algo version2). The model is built upon Darknet YOLO but is also ported on Tensorflow Lite, Protobuf file and also in YAD2K (Keras port of YOLO).

Primary LanguageJupyter Notebook

Weapon Detection

  • Weapon Detection Deep Learning model built using YOLO (You Only Look Once Algorithm)
  • This was only trained for guns at an extent.
  • The model was trained using tiny-yolov2
  • It was also converted to protobuf, yad2k and tflite formats.

All the weights can be found here

Used Library

Darknet YOLO

Implementation can be found here!