- Make sure you have an NVidia graphics card which supports CUDA
- Install the CUDA drivers
- Fix requirements.txt to add missing dependencies
- pip install -e .
- Re-install pytorch using instructions from their website
- Run
flask run --no-reloadto prevent app from restarting while running detection
Tomato image 17 Pear image 12 Banana image 1 Onion image 1 Carrot image 1