当前位置:   article > 正文

【存档记录】AutoDL云服务器训练yolov5coco128数据_yolov5总下载coco128.zip

yolov5总下载coco128.zip

AutoDL云服务器训练yolov5coco128数据


镜像选择:PyTorch 1.7.0
Python 3.8(ubuntu18.04)
Cuda 11.0
开机后:终端命令行输入

环境配置部分

source activate
conda create -n yolov5coco python=3.8
conda activate yolov5coco
pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

代码及数据集处理部分

  1. 下载代码并安装相关文件

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt

  1. 下载数据集coco128.zip解压到yolov5目录下
    数据集来源http://t.csdn.cn/sCqgm
    tips:一个小坑coco128.yaml文件需要修改
path: ../datasets/coco128  # dataset root dir
  • 1

autodl默认…/是从root开始算,这样导致找不到已下载的数据集,改为

path: ../yolov5/datasets/coco128  # dataset root dir
  • 1

开始训练

python train.py
完成

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/149690
推荐阅读
相关标签
  

闽ICP备14008679号