1.创建虚拟环境
conda env list
conda create -n yolov5 python==3.9
conda activate yolov5
2.安装torch和torchvision
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
3.下载项目
https://github.com/ultralytics/yolov5/archive/refs/tags/v7.0.zip
4.准备数据集
Vehicles-OpenImages Object Detection Dataset - 416x416
选择yolov5格式
5.训练模型
python train.py --data ../MaskDataSet/data.yaml --cfg models/yolov5s.yaml --batch-size 16
11万+

被折叠的 条评论
为什么被折叠?



