预训练权重

参考官方的yolov5

准备数据集

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

输出为shp形式。

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
data		data
inference		inference
models		models
utils		utils
README.md		README.md
config_test.txt		config_test.txt
config_train.txt		config_train.txt
config_train2.txt		config_train2.txt
detect.py		detect.py
hubconf.py		hubconf.py
prepare_data.py		prepare_data.py
readme.txt		readme.txt
test.py		test.py
test_concate.py		test_concate.py
train.py		train.py