预训练权重 参考官方的yolov5 准备数据集 标签为shp形式，新建字段“cls”, 里面填写类别；一个标签和对应的影像为一组，放置在一个文件夹内，参考config_train.txt第一行。设置完成后，直接运行train.py。 模型预测 输出为shp形式。