This is a simple GUI using PyQt for YOLO darknet mass detection NeuralSIGHT from final bachelor capstone project.
- PyQt5
- YOLO darknet installed
- OpenCV >= 4.4.0
Follow AlexeyAB's instruction of How to compile YOLO darknet on Linux or Windows
Simple installation from PyPI
pip install PyQt5
pip install opencv-python
From Conda
conda install -c anaconda pyqt
- Move
main.py
,main_window.py
, andsplashscreen.py
to your darknet folder. - Put your
your-weight.weights
to weights folder.
if you're going to use GUI with OpenCV skip to Step 3, otherwise follow these instructions for YOLO version inference
- Edit
main_window.py
on line 190 to your darknet data folder directory path. - If your OS is Windows or you have different name for your own cfg or weights files then edit
main_window.py
on line 200 todarknet.exe detector test cfg/[your configuration].data cfg/[your configuration].cfg weights/[your weights].weights data/data.jpg -dont_show
Run main.py
for YOLO version inference
Run mainCV.py
for OpenCV version inference