Requirements

This assumes you are running ubuntu 16.04 and already have darknet/yolo installed. The regular version of darknet/yolo can either be downloaded by following the instrucitons here, or from the westpoint-robotics repo here. For the ROS integrated yolo, our local repo is here.

Installation of threat detection module

mkdir -p ~/threat_detection/azure/rifle && cd ~/threat_detection

git init && git remote add gh git@github.com:westpoint-robotics/threat_detection.git && git pull gh master

Datasets

Download our dataset

Download here

Creating a new dataset

To download new images for making a new dataset follow instructions here

To label new class, follow instructions here

Usage

TBD. We are still working on this part.

Ros packages to install:

https://github.com/benjaminabruzzo/stop
https://github.com/westpoint-robotics/usma_ardrone
https://github.com/westpoint-robotics/darknet-yolov3-ros

Testing weights and config using ROS and AR.Drone Camera:

roslaunch optitrack_controller ardrone.launch
roslaunch darknet_ros yolo_pistols.launch

Downloading pose caffemodels

cd ~/threat_detection && . /pose_detection/pose_models/getModels.sh 

Using virtual environments for python

sudo pip install virtualenv virtualenvwrapper
sudo rm -rf ~/.cache/pip

echo 'export WORKON_HOME=$HOME/.virtualenvs' >> ~/.bashrc
echo 'source /usr/local/bin/virtualenvwrapper.sh' >> ~/.bashrc

mkvirtualenv cvn


cd ~ && git clone git@github.com:opencv/opencv.git && cd opencv && git checkout 3.0.0

mkdir -p ~/opencv/opencv/build
cd ~/opencv/opencv/build
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D INSTALL_C_EXAMPLES=OFF -D INSTALL_PYTHON_EXAMPLES=ON -D OPENCV_EXTRA_MODULES_PATH=~/opencv_contrib/modules -D BUILD_EXAMPLES=ON ..


https://nbviewer.jupyter.org/github/westpoint-robotics/threat_detection/blob/master/detect_threat_level/Skeleton%20Threat%20Detection.ipynb



ssh 

scp user1@10.113.1.252:/home/user1/Carey/kevin_yolo/alt/darknet/people_pistol/data/Images/001 ~/datasets/iggy/001