pycharm运行yolov5 环境配置

主要是备注踩过的一些坑,这里使用的是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))
发布日期:
分类:资料

发表评论