数据集准备 在项目代码平级路径下创建数据集路径,包括.yaml的数据集配置文件 训练集,验证集,测试集路径如下: 包括图像和标注,图像和标注对应文件名相同 标注为.txt的yolo格式如下: 类别 x y w h (坐标值为归一化后的相对值) .yaml的数据集配置文件如下: 训练模型 环境:python3.6 + torch1.6.0 + cu101 + CUDA10.0 + cudnn7.6.0 在train.py中指定配置参数