【学习记录】
一、安装环境:
【yolov7环境中完全可以】,直接安装requirement.txt即可
1、创建虚拟环境
conda create -n yolov7 python=3.7 -y
conda activate yolov7
#退出环境
conda deactivate
2、下载源码:
git clone https://github.com/ultralytics/ultralytics
cd ultralytics
pip install -e '.[dev]'
3、安装依赖
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
【注意】训练yolov7时已经单独安装过pytorch
二、训练:
1、下载预训练模型
2、训练
1:yolo task=detect mode=train model=yolov8x.yaml data=mydata.yaml epochs=350 batch=8
3、预测
yolo task=detect mode=predict model=yolov8n.pt device=0 show=True(实时显示)
【训练时遇到的问题】
问题1:CUDA error:out of memory
解决:nvidia-smi #查看gpu使用情况
kill -9 -PID #杀掉没用的进程
问题2:tktinker会报错,按照网上的来就行
解决:报错的文件中:from mttkinter import mtTkinter as tk