Skip to content

Latest commit

 

History

History
executable file
·
28 lines (18 loc) · 579 Bytes

README.md

File metadata and controls

executable file
·
28 lines (18 loc) · 579 Bytes

DIDN

Dangerous Item Detection Network based trained with YOLOv3 and translated to Tensorflow library with usage of tool used for translation

Folder structure:

  • model - space to place trained model
  • test_set - sample video for testing

Installing dependencies

Pip

# TensorFlow CPU
pip install -r requirements.txt

# TensorFlow GPU
pip install -r requirements-gpu.txt

Running

Detection of weapon from sample video

python detect.py