预训练权重

参考官方的yolov5

准备数据集

标签为shp形式，新建字段“cls”, 里面填写类别；一个标签和对应的影像为一组，放置在一个文件夹内，参考config_train.txt第一行。设置完成后，直接运行train.py。

模型预测

输出为shp形式。