跳至内容

2025-11-29 视觉培训记录

标数据集

1
 conda create -n yolov5 python=3.11

配环境

1
conda activate yolov5

torch

Get Started

1
2
3
pip3 install torch torchvision #或者下面的

pip3 install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
1
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
1
pip install tensorboard

运行data_organize.py分为训练集,测试集

写yaml文件

1
yolo train data=mydata.yaml model=yolov5s.pt epochs=300 imgsz=640 batch=8 workers=0 device=0