文章目录
- 软硬件准备
- 数据准备
- 数据处理脚本
- 模型训练
- 模型部署
- 数据分享
软硬件准备
训练端
- Pytorch
- ultralytics
- Nvidia 3080Ti
部署端
数据准备
用labelimg进行数据标注
数据处理脚本
xml2yolo
import os
import glob
import xml.etree.ElementTree as ETxml_file='/home/nvidia/yp/training/grape/xml'
txt_file = "/home/nvidia/yp/training/grape/txts"l
