你要知道训练的门槛是什么?前面说不能训练的也是以一定标准为前提 1.首先yolov5batch size设成1,图片输入使用默认输入看看显存会不会爆 2.1如果不行,修改图片大小 3.修改网络结构,选择yolov5的变种,或者更换backbone 这台gpu只有4G显存,但理论上调到最低是可以的 只是...
只是学习性质的话,可以拿yolov5-s玩一玩,input改到320*320或者更低,应该可以整个4-8的bs。拿来做...
我batch-size设置为4,epoch为50,数据集总共一万多的照片跑一下九个多小时,虽然慢但是最后训练出来也可以用 只是