yolo3-pytorch-master源码下载地址:GitHub - bubbliiiing/yolo3-pytorch: 这是一个yolo3-pytorch的源码,可以用于训练自己的模型。
首先安装anaconda和pycharm软件,网络上相关博客很多,都可以参考。
然后在pycharm中打开yolo3-pytorch-master项目,在右下方的环境中单击选择Add Interpreter...选项,在弹出的界面上选择Conda Environment,在右边选择环境存储位置,一般是anaconda下的envs文件夹 + 环境名;随后选择你所需要的python版本,最后勾选是否想将该环境应用于所有项目,点击 确定 创建环境。
环境创建成功后根据需求安装相应的第三方库
打开下方的Terminal
切换我们刚刚创建的环境,-name为刚创建成功的环境名
activate -name
1.安装torch==1.2.0和torchvision==0.4.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
2. 安装scipy==1.2.1、numpy==1.17.0、matplotlib==3.1.2、Pillow==8.2.0、h5py==2.10.0
pip install scipy==1.2.1 numpy==1.17.0 matplotlib==3.1.2 Pillow==8.2.0 h5py==2.10.0
3. 安装opencv_python==4.1.2.30
pip install opencv_python==4.1.2.30
4.另外在summary.py中有第三方库torchsummary,虽然该部分代码仅用于看网络结构,可看需求安装。安装时需要注意,该库只能使用pip安装,在conda里是没有这个包的
pip install torchsummary