主要是备注踩过的一些坑,这里使用的是5.0版本。
首先要安装pytorch,然后再安装yolov5的依赖库,因为直接安装的话yolov5安装的torch是cpu版本的训练。
pycococtools安装补上的问题:pycococtools的安装包https://pan.baidu.com/s/1nWQdPRtGwNnOO2DkxRuGuA提取码:i5d7
或者安装vs最新版以后
google-auth 报错,删除干净,然后删除tensorboard,重新安装
pip install –upgrade google-auth
pip install sensorboard
报错:AttributeError: Can’t get attribute ‘SPPF’ on <module ‘models.common
把下面的代码直接复制到common.py最下面就OK了
def forward(self, x):
x = self.cv1(x)
with warnings.catch_warnings():
warnings.simplefilter('ignore') # suppress torch 1.9.0 max_pool2d() warning
y1 = self.m(x)
y2 = self.m(y1)
return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))