1.下载yolov5模型
github官方网址如下:GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
选择版本,这里使用的是yolov5的第五个版本,截止到写笔记的时候已经更新到第七个版本了,但是第七个版本需要python版本>=3.9,而我之前配的pyt环境的python版本为3.8,为了能直接使用之前配好的pyt环境,这里选择的是第五个版本。
2.打开项目并配置package
在pycharm中打开这个项目,并且设置环境为之前配好的pyt,也可以新建一个环境,一个个安装它需要的包。
在setting框中,点击project:detect.py,下拉框中点击python Interpreter,在这个界面中可以选择python interpreter,选择之前在anaconda中配好的pyt,然后点击APPLY,在点击OK
如果是新建环境,就在打开的时候选择它自己给你推荐的python 3.11那个环境,然后省略上面的那一步,因为后面的步骤是一样的。
yolov5项目带了一个名叫requirements.txt的文件,里面写明了运行yolov5所需要的包和版本,可以直接在terminal里面输入pip install -r requirments.txt安装里面的所有包
之前选择自己配好的pyt环境是因为之前配的环境已经安装了其中的大部分内容了,就不会再重复安装了,如果是用一个新的环境来安装里面的所有包,还是需要一点时间的。
值得注意的是,打开项目的时候要注意requirments.txt在根目录下,也就是说在yolov5-5.0这个文件夹为根目录的前提下打开,如果一不小心在前几个文件夹下打开的目录,那么这个相对路径就不对了,直接把requirments.txt改成正确的相对路径就行了。
3.下载权值文件
权值文件yolov5(s/m/l/x).pt 与yolov5(s/m/l/x)6.pt八种,从s到x分别代表模型的大小,依次是模型从小到大,训练效果从小到大。这些权值文件存储在根目录下,但是我们下载的这个项目是不带这些文件的,如果根目录下没有所需要的模型的文件是会自动从github上下载,但是它会下载最新的那个版本的模型的权值文件,比如下载官网上已经更新到了7.0版本,他会自动下载yolov5-7.0对应的权值文件,所以会报错,我们自己在官网上下载了复制到根目录下即可。