johnolafenwa / deepstack-trainer

Custom Object Detection Training for DeepStack
GNU General Public License v3.0
20 stars 12 forks source link

Add inference step after training #7

Open robmarkcole opened 3 years ago

robmarkcole commented 3 years ago

As its always nice to see. In my case performed with:

!python /content/deepstack-trainer/detect.py --weights /content/deepstack-trainer/train-runs/exp/weights/best.pt --img 640 --conf 0.25 --source /content/valid/images/

This batch processes all the images in /content/valid/images/. Then the images are in a directory /content/runs/detect/exp