封装程序:pyinstaller -F (文件名)
1、模型训练:
从零开始训练:python train.py --data .\data\TianMao_Vertical.yaml --cfg yolov5l.yaml --weights .\yolov5l.pt --batch-size 14 --epochs 200 --img-size 640 --workers 2
从exp37开始叠加训练:python train.py --data .\data\HongTa_Vertical.yaml --cfg yolov5l.yaml --weights .\runs\train\exp37\weights\best.pt --batch-size 8 --epochs 300 --img-size 640 --workers 0
检验模型图片:python .\detect.py --weights .\runs\train\exp7\weights\best.pt --img-size 640 --source ..\dataset\HongTa_Vertical\images\train\
pt文件转换成wts文件:python .\gen_wts.py -w .\runs\train\exp41\weights\best.pt -o .\runs\train\exp41\weights\best.wts
pt文件转换成onnx文件:python .\onnx_model_export\export.py --weights .\runs\train\exp112\weights\best.pt --img 640 --batch 1
2、(Linux)安装cuda,会自动安装显卡驱动(nvidia-driver)
(1)禁用ubuntu自带的nouveau驱动(第三方显卡驱动)
sudo gedit /etc/modprobe.d/blacklist_nouveau.conf
系统会打开一个空白文件,在里面写入以下两行:
blacklist nouveau
options nouveau modeset=0
然后保存关闭
(2)在终端输入$reboot重启电脑
(3)CUDA-toolkit install guide
我们可以从NVIDIA官网CUDA下载页面,网址为https://developer.nvidia.com/cuda-toolkit-archive ,点击CUDA Toolkit 11.7.1(选择不高于nvidia-smi提示的最高版本cuda)
如图
(4)请使用deb local方式安装来install CUDA和TensorRT, 别使用run方式安装
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.1-515.65.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.1-515.65.01-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt update 或 $sudo apt-get update
sudo apt-get -y install cuda
离线安装CUDA Toolkit 11.7.1:
sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.1-515.65.01-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt update 或 $sudo apt-get update
sudo apt-get -y install cuda
安装cudatoolkit的同时,会自动安装显卡驱动(nvidia-driver)
(5)终端输入nvidia-smi,查看是否有打印信息
例如:NVIDIA-SMI 470.103.01 Driver Version: 470.103.01 CUDA Version: 11.4
(6)查看cuda是否被安装
dpkg -l | grep cuda
//类似以下输出说明安装成功
...
ii cuda 11.4.4-1 amd64 CUDA meta-package
ii cuda-11-4 11.4.4-1 amd64 CUDA 11.4 meta-package
ii cuda-cccl-11-4 11.4.122-1 amd64 CUDA CCCL
ii cuda-command-line-tools-11-4 11.4.4-1 amd64 CUDA command-line tools
ii cuda-compiler-11-4 11.4.4-1 amd64 CUDA compiler
ii cuda-cudart-11-4 11.4.148-1 amd64 CUDA Runtime native Libraries
(7)在终端输入reboot重启电脑
3、下载 pycharm(版本:PyCharm Community)
官网:Download PyCharm: Python IDE for Professional Developers by JetBrains
Linux安装操作网址:https://blog.csdn.net/weixin_44906084/article/details/125037653
4、Anaconda3安装
Linux安装操作网址:Linux 系统下安装 anaconda3-物联沃-IOTWORD物联网
(1)配置环境变量(Linux):终端输入:gedit ~/.bashrc
在文档最后添加变量:export PATH=/home/weilai/anacond3/bin/:$PATH
(2)进入anaconda环境:source ~/anaconda3/bin/activate
(3)查看anaconda安装的环境:conda info --env或conda env list
(4)进入创建的虚拟环境: activate (环境名)
(5)anaconda中创建虚拟环境: conda create -n (新环境名) --clone (旧环境名)
(6)使用最新版本的Python3创建环境。我们将环境称为my_env:$ conda create --name my_env python=3.7
(7)查看python版本:python -V或python --version
(8)升级python版本:conda update python
(9)激活虚拟环境:conda activate env_name(环境名称)
(10)退出虚拟环境:deactivate env_name(环境名称)
(11)删除虚拟环境:conda remove -n env_name(环境名称) --all
(12)安装包:conda install [package_name[=version]
(13)搜索哪些Python版本可供使用:$ conda search "^python$"
(14)为yolov5安装需要的包:pip install -r [package_name]
先在conda需要的base环境进入yolov5的文件夹:cd /home/user/yolov5-5.0
输入命令:pip install -r [package_name]
(15)添加源:源前面加 -i
清华:https://pypi.tuna.tsinghua.edu.cn/simple
阿里云:https://mirrors.aliyun.com/pypi/simple
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学:http://pypi.hustunique.com/
山东理工大学:http://pypi.sdutlinux.org/
豆瓣:https://pypi.douban.com/simple/
安装解压包:sudo dpkg -i [package_name]
查看资源占用:watch -n 1 nvidia-smi
5、NVIDIA官网
CUDA Deep Neural Network (cuDNN) | NVIDIA Developer
6、linux安装中文输入法 终端输入:sudo apt-get install ibus-pinyin
在设置里面“区域与语言”,“中文(智能拼音)”
Linux操作网址:在linux系统下如何下载中文输入法,如何在linux系统下安装中文输入法_CharlesxxxHong的博客-CSDN博客
7、pytorch安装(需要对应版本安装)
查看版本网址:https://pytorch.org/get-started/previous-versions/
下载离线安装网址:https://download.pytorch.org/whl/torch/