Name		Name	Last commit message	Last commit date
parent directory ..
model		model
test_set		test_set
.dockerignore		.dockerignore
Dockerfile		Dockerfile
README.md		README.md
detect.py		detect.py
requirements-gpu.txt		requirements-gpu.txt
requirements.txt		requirements.txt

README.md

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