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