PyTorch安装

104 阅读1分钟

因为工作需要学习下目标检测实现。实现口罩、安全帽、烟火识别 自己记录下搭建yolov5环境过程 YOLOv5是一种单阶段目标检测算法,该算法在YOLOv4的基础上添加了一些新的改进思路,使其速度与精度都得到了极大的性能提升。主要的改进思路如下所示:

输入端:在模型训练阶段,提出了一些改进思路,主要包括Mosaic数据增强、自适应锚框计算、自适应图片缩放; 基准网络:融合其它检测算法中的一些新思路,主要包括:Focus结构与CSP结构; Neck网络:目标检测网络在BackBone与最后的Head输出层之间往往会插入一些层,Yolov5中添加了FPN+PAN结构; Head输出层:输出层的锚框机制与YOLOv4相同,主要改进的是训练时的损失函数GIOU_Loss,以及预测框筛选的DIOU_nms。

1、安装显卡驱动(www.nvidia.cn/Download/in…

image.png

image.png 2、anaconda安装(www.anaconda.com/products/di…

image.png

image.png

image.png

安装完毕之后请务必更新为国内的源,加快第三方库的下载速度,在cmd中执行下列指令即可。

conda config --remove-key channels

conda config --add channels mirrors.ustc.edu.cn/anaconda/pk…

conda config --add channels mirrors.ustc.edu.cn/anaconda/pk…

conda config --add channels mirrors.bfsu.edu.cn/anaconda/cl…

conda config --set show_channel_urls yes

pip config set global.index-url mirrors.ustc.edu.cn/pypi/web/si…

 

3、PyTorch安装(pytorch.org/)

image.png 参考:

blog.csdn.net/ECHOSON/art…

blog.csdn.net/zhouzhiyao9…

blog.csdn.net/didiaopao/a…