YOLO v5 recognition

Stuff I am working on
Daniel Wee
Site Admin
Posts: 2449
Joined: Wed 25 Feb 25 2009 8:00 pm

Re: YOLO v5 recognition

Post by Daniel Wee »

Attempting to retrain with the Windows 10 setup:-

Code: Select all

python train.py --img 640 --batch 16 --epochs 300 --data training2/data.yaml --cfg training2/yolov5l.yaml --weights ''
Results: FAIL

There must be some fundamental problem preventing training on Windows 10. Inference works though.
Daniel Wee
Site Admin
Posts: 2449
Joined: Wed 25 Feb 25 2009 8:00 pm

Re: YOLO v5 recognition

Post by Daniel Wee »

We can further speed up inference if TensorRT cores are available (NVIDIA V100-A100 GPUs). This is especially useful for implementations on the TX2 or AGX (or even NX) boards.

https://github.com/wang-xinyu/tensorrtx ... ter/yolov5

Further information here:-

https://www.learnopencv.com/how-to-conv ... inference/
Post Reply