- export目录的作用是为训练好的模型进行模型转换,以便可以部署
- export.sh是训练后一键转换模型的bash脚本
- tools目录的作用是为训练前的数据集进行预处理,以便可以训练
- train目录的作用是进行模型训练
- train.sh是一键执行训练的bash脚本
- pt后缀的都是训练需要的预训练模型,是官方训练效果很好的模型
- dataset是运行train.sh根据用户给的数据集构建的可训练的目录,与用户指定的路径互不影响
- 一键训练官方yolov8的目标检测或者实例分割模型
- 只需运行bash脚本,根据提示输入即可训练:
bash train.sh
- 运行环境:
docker pull registry.cn-shanghai.aliyuncs.com/epo0408/yolov8_ocr:v1.1
- images是划分好的图片目录
- json是所有图片的json标注文件
- labels是用于训练的由json转换而来并且划分好的txt文件
- tmp里存放那个是所有图片的地址
- yaml文件是自动生成的含有训练集、验证集以及类别信息的训练配置文件
-
训练完毕的best.pt文件会在上面说的存放训练文件的目录下找到。
-
只需要执行:
bash export.sh 模型pt文件的绝对路径
即可转换模型,转好的模型和类别码与best.pt同级