预训练权重

参考官方的yolov5

准备数据集

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

输出为shp形式。

Name	Name	Last commit message	Last commit date
Latest commit SonwYang Add files via upload Mar 15, 2022 1850c82 · Mar 15, 2022 History 2 Commits
data	data	first commit	Nov 5, 2020
inference	inference	first commit	Nov 5, 2020
models	models	first commit	Nov 5, 2020
utils	utils	first commit	Nov 5, 2020
README.md	README.md	first commit	Nov 5, 2020
config_test.txt	config_test.txt	first commit	Nov 5, 2020
config_train.txt	config_train.txt	first commit	Nov 5, 2020
config_train2.txt	config_train2.txt	Add files via upload	Mar 15, 2022
detect.py	detect.py	first commit	Nov 5, 2020
hubconf.py	hubconf.py	first commit	Nov 5, 2020
prepare_data.py	prepare_data.py	first commit	Nov 5, 2020
readme.txt	readme.txt	first commit	Nov 5, 2020
test.py	test.py	first commit	Nov 5, 2020
test_concate.py	test_concate.py	first commit	Nov 5, 2020
train.py	train.py	first commit	Nov 5, 2020