全球人工智能技术创新大赛之布匹疵点智能识别 task1

1.程序代码跑通

1.1 代码结构

baseline代码
代码结构

代码跑通步骤：

• 下载数据集放入train_data文件夹中。
• 训练模型。
• 测试模型性能。

1.2 step1：下载数据集

根据官方给出的guangdong2_data_list.txt中的链接下载好

• guangdong1_round2_train2_20191004_Annotations
• guangdong1_round2_train2_20191004_images
  把他们解压后原封不动放入train_data中。

1.3 step2：训练网络

第一步，处理数据。

python convertTrainLabel.py 
python process_data_yolo.py 

注意process_data_yolo.py只能转换val数据集，所以将代码中的val都改成train来转换train数据集。
第二步，训练网络。
下载预训练权重yolo5s.pt，链接点我。将其放置在weights文件夹下，更改名字为best.pt。
注意将train代码中的第120行代码进行更改。

# 更改前
if model.state_dict()[k].shape == v.shape}
# 更改后
if k in model.state_dict() and model.state_dict()