分割出城市中的垃圾堆场,遂寻找到了 UrbanAtlas 2012 database 这个数据集和对应的 baseline
baseline
IoU for class 1: 0.5667
IoU for class 2: 0.3504
IoU for class 3: 0.0001
IoU for class 4: 0.0002
IoU for class 5: 0.4121
IoU for class 6: 0.0204
IoU for class 7: 0.2439
IoU for class 8: 0.0000
IoU for class 9: 0.0000
IoU for class 10: 0.5839
IoU for class 11: 0.1610
IoU for class 12: 0.0241
IoU for class 13: 0.0161
IoU for class 14: 0.1747
Average IoU: 0.1824
基于 baseline 的改动
阅读了 baseline 的全部源码,发现了一些漏洞吧:
- 图像大小是 2000X2000 的,baseline 只随机选取其中的 512X512 进行训练,遂切割数据重新制作数据集
- 仔细观察每个类别,发现相同类