Utilizes the YOLO classifying algorithm to identify possible targets.
-> See https://github.com/jromie0924/turret-GPIO for the Arduino code.
It's worth noting that the YOLO library only works on Linux. This has been written and tested on Ubuntu 20.04.
NOTE: This requires Python 3.8, as well as an NVIDIA GPU that has CUDA cores.
sudo pip3 install pipenv
- Activate pipenv virtual environment (
pipenv shell) python -m pip install opencv-python- Test that it works:
python -c "import cv2; print(cv2.__version__)"
pipenv install
pipenv shell(if you're not already inside the virtual environment)python app.py
Note that it's helpful if you have a webcam plugged into your machine; the program will likely not like it if you don't.
You may run into an issue with yolo34py-gpu package.
sudo apt updatesudo apt install nvidia-cuda-toolkitgit clone [email protected]:AlexeyAB/darknet.gitcd darknet- Open the
Makefileand set theGPUvariable to1. Save it. make- Change directories back to the turret repo and run
pipenv installagain.
You may get a "permission denied" error with serial connection. Add your user to the dialout user group:
sudo usermod -a -G dialout $USER