ubuntu 16.04 conda dependencies: see environment.yml cuda: 9.2
https://drive.google.com/file/d/1yD8FOPCnh3q-j7pekfab4pfkV8t3dN2x/view?usp=sharing
put weights in the root
python CYWMain.py
or
python3 CYWMain.py\
###1. Stereolabs ZED - YOLO 3D in Python
This package lets you use YOLO the deep learning object detector using the ZED stereo camera in Python 3.
The left image will be used to display the detected objects alongside the distance of each, using the ZED Depth.
- Windows 7 64bits or later, Ubuntu 16.04
- ZED SDK and its dependencies (CUDA)
- ZED Python 3 wrapper
Download and install the ZED Python wrapper following the instructions, to make sure everything works you sould try a sample.
We will use a fork of darknet from @AlexeyAB : https://github.com/AlexeyAB/darknet
- It is already present in the folder libdarknet
- Simply call make in the folder
cd libdarknet
make -j4
- For more information regarding the compilation instructions, check the darknet Readme here
- Download the model file, for instance Yolov3 tiny
wget https://pjreddie.com/media/files/yolov3-tiny.weights
To launch the ZED with YOLO simply run the script :
python3 darknet_zed.py
The input parameters can be changed using the command line :
python3 darknet_zed.py -c <config> -w <weight> -m <meta> -t <threshold> -s <svo_file>
For instance :
python3 darknet_zed.py -c cfg/yolov3.cfg -w yolov3.weights -m cfg/coco.data -t 0.5
To display the help :
python3 darknet_zed.py -h