一、参考博客和注意问题
一、参考博客:
pytorch训练自己的YOLOv5目标检测器(自定义数据集训练)
WIN10下YOLOV5安装与测试-踩坑记会发现我们下载的requirements.txt与博主的不一样,源码发生了变化注意:不要采用 pip install -U -r requirements.txt原因:应该只支持Linux和macos系统,对于window系统还是采用下面的操作进行配置。
二、 安装Anaconda、安装VS2015、安装CUDA10.2.89、CUDNN7.6.5(操作看我的Tensorflow object detection api(maskrcnn的搭建流程))
三、创建虚拟环境:conda create -n yolov5 python=3.7
四6.1为conda设置清华源:打开Anaconda Prompt,输入清华仓库镜像,这样更新会快一些:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
6.2使用 pip install 时,可以在代码最后附上以下清华镜像源代码加快下载速度:
-i https://pypi.tuna.tsinghua.edu.cn/simple
五、在虚拟环境安装cuda和cudnn和安装各种库:
conda install cudatoolkit==10.2.89
conda install cudnn==7.6.5
pip install numpy==1.17 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install tensorboard -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install pyyaml -i https://pypi.tuna.tsinghua.edu.cn/simple
# pip install git -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install pandas -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install Cython -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
六 、pycocotools安装,下载连接:,并解压;
百度网盘包:
链接:https://pan.baidu.com/s/1J3O-VmnRrYqNBEEpEXAo6A
提取码:qqmu
以管理员身份打开 CMD 终端,并切换到 cocoapi\PythonAPI目录。运行以下指令:
cd D:\software\CNN_package\cocoapi-master\PythonAPI
d:
# install pycocotools locally
python setup.py build_ext --inplace
# install pycocotools to the Python site-packages
python setup.py build_ext install
七、安装pytorch 1.5.1
pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
# pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
九、1.1.1 :下载yolov5源码
十、1.1.2. 下载yolov5模型(.pt文件)
十一、测试:
cd E:\yolov5
e:
python detect.py --source ./inference/images/ --weights ./weights/yolov5s.pt --conf 0.4
python detect.py --source 0
原文链接:https://blog.csdn.net/qq_41627642/article/details/107628727