macrobomaster / cv-yolov7

Custom yolov7 model with opencv
GNU General Public License v3.0
5 stars 3 forks source link

Try different models for better fps #5

Closed Bill-Haoyu-Lin closed 1 year ago

Bill-Haoyu-Lin commented 1 year ago

Try different yolov5-tiny other model with small dataset, since we need a smaller model for NvidiaTX2 developer board.

Bill-Haoyu-Lin commented 1 year ago

Used yolov5s for smaller model try. Fps increase but classifier need to be improved for yolov5

Bill-Haoyu-Lin commented 1 year ago

tried yolov5n for 30 + fps in stable condition and max of 58 fps, MicrosoftTeams-image

Bill-Haoyu-Lin commented 1 year ago

Need to train model to fit on more various data ( Armor )

We will also need to input the data for labels ( letters around the battle ground ), image

We will still be using tiny model for this task, may move to custom model without yolo suite