YOLOv3学习准备

  本次软件杯选题选择的是A4,关于林业害虫的识别,在算法上我想尝试一下YOLOv3,因此先搭建一下环境。

  操作系统:Win10(后面小组会租用服务器,学习先用本机跑着)

  环境准备:Anaconda3

  参考链接:官网链接:

https://github.com/ultralytics/yolov3(YOLOv3)

https://pytorch.org/get-started/locally/(Pytorch)

  首先我们看一下YOLOv3的官方文档上对环境有什么要求:

 

   要求Python3.8,torch版本>=1.7,最后再用pip安装requirements.txt内的东西。打开Anaconda创建虚拟环境,可以选择可视化界面操作,我这里直接用命令行窗口了,代码如下:

conda create -n pytorch python=3.8

 

 先看看requirements.txt里都有什么:

 

 这里可以看到它里面预写了torch,如果直接运行的话也能配置好pytorch的,但我不确定这个pytorch有没有装CUDA,我的电脑有NVIDIA显卡,所以想装GPU的,因此pytorch我选择自己装,下面去看看pytorch官网:

 

 这里没有选择conda的原因是第一次安装时报了HTTP000的错误,但镜像源我是配置好了的,实在不明白原因我就改用了pip,pip我忘了设置镜像源,因此下的时候很慢,可以选择后面接"-i + 源"临时使用或者更改pip的源。

下载好之后我们进入python尝试一下

 

 没有报错,证明安装成功,接下来我们把reqquirements.txt下载到本地,记住位置,然后使用命令开始安装需要的包:

 

 我这里直接把yolov3所有的源代码下载过来了,git clone或者其他的方法都可以。

回到命令行窗口,进入到txt所在位置,运行pip:

 

 这里写错了个字母(大雾)

 

 我这里是已经安装好了,会提示这个,没有的话就要慢慢等了。

友情提示:下载的时候建议后面使用 -i 临时使用国内源,或者更改好pip的源,否则下载很慢。

问题记录:使用了Python3.7下载pytorch,当把python版本更新到3.8的时候,pytorch无法使用。

解决方法:删除pytorch,在py3.8的情况下重装。

posted @ 2021-04-25 17:42  千幽行  阅读(194)  评论(0编辑  收藏  举报