Thanks to visit codestin.com
Credit goes to github.com

Skip to content

jromie0924/turret

Repository files navigation

TURRET

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.

Setup

NOTE: This requires Python 3.8, as well as an NVIDIA GPU that has CUDA cores.

Install pipenv

sudo pip3 install pipenv

Install OpenCV for Python

  1. Activate pipenv virtual environment (pipenv shell)
  2. python -m pip install opencv-python
  3. Test that it works: python -c "import cv2; print(cv2.__version__)"

Install other packages

pipenv install

Running

  1. pipenv shell (if you're not already inside the virtual environment)
  2. 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.

Troubleshooting

Problem 1

You may run into an issue with yolo34py-gpu package.

  1. sudo apt update
  2. sudo apt install nvidia-cuda-toolkit
  3. git clone [email protected]:AlexeyAB/darknet.git
  4. cd darknet
  5. Open the Makefile and set the GPU variable to 1. Save it.
  6. make
  7. Change directories back to the turret repo and run pipenv install again.

Problem 2

You may get a "permission denied" error with serial connection. Add your user to the dialout user group:

  • sudo usermod -a -G dialout $USER

About

Tracks objects

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages