[日常填坑]图像分类实战-训练top1和top3(网络模型源码已经建好)

下载数据集

top3
1 组织测试数据,将所有的测试图片放到同一个文件夹下

2 更改./config.py中test_path为该文件夹地址,形如'/home/ubuntu/test'。更改phase为 'test'
3 将./utils/test.py最后一行改为 writer.writerow([name, top3[0], top3[1], top3[2]]

4 创建test_result.csv文件,并在文件的第一行加上 FILE_ID,CATEGORY_ID0,CATEGORY_ID1,CATEGORY_ID2 然后保存
5 更改./config.py中test_csv 为上述test_result.csv地址
6 在项目根目录下运行 python main.py

7 获取csv文件(注意将csv格式设置正确,最好为utf-8)
8 参赛者须提供每张图片最有可能的3个分类的ID,按置信度降序排列
测试结果通过独立的CSV文件提交。格式如下:
FILE_ID,CATEGORY_ID0,CATEGORY_ID1,CATEGORY_ID2
47c2x4b41c4011e8aaec00163e025683,3,5,8
32c2x4b41c4011e8aaec00163e025683,2,5,8
71c8x6b41c4011e8aaec00163e025683,2,3,5
csv字段使用半角,分隔,换行符使用\n

----

top1

1 组织测试数据,将所有的测试图片放到同一个文件夹下
2 更改./config.py中test_path为该文件夹地址,形如'/home/ubuntu/test'。更改phase为 'test'
3 将./utils/test.py最后一行改为 writer.writerow([name, top1[0]]
4 创建test_result.csv文件,并在文件的第一行加上 FILE_ID,CATEGORY_ID0然后保存
5 更改./config.py中test_csv 为上述test_result.csv地址
6 在项目根目录下运行 python main.py

 

猜你喜欢

转载自www.cnblogs.com/hoanfir/p/9047670.html