DingXiaoH / RepVGG

RepVGG: Making VGG-style ConvNets Great Again
MIT License
3.31k stars 434 forks source link

questions about training on custom datasets? #80

Open Hezhexi2002 opened 2 years ago

Hezhexi2002 commented 2 years ago

大佬,我就是之前在知乎上给您私信黑盒本质的同学,现在在中北大学卓创RM战队视觉组,然后主要是负责用神经网络去检测装甲板信息,说白了就是目标检测,但是主要就是对于速度要求比较高,因为实验室有个nano和nuc,想的是把模型部署到nano上进行推理,但是目前我已经试过了darknet版的yolov4-tiny在我笔记本rtx 2060上只能65FPS左右,然后在nano上也就十几帧了,试过用trt加速,也才25FPS左右,之后又陆续试过了yolov5s,yoloxs,其中yolov5s在2060上能做到7ms左右,但yolox也得20ms左右,然后目前效果最好的是u版的yolov3-tiny能到4ms左右(未经过trt加速),我看您的repvgg好像对于推理很友好,但是您的仓库里对于train custom dataset介绍的很少,就是我想问一下数据集该怎样组织和加载呢?[思考]