未经允许,不要随意转载!
最近想试试用FCN训练自己的数据集,想想标注是一件很费劲的事,于是就想到了用其他库里的数据,这里选用了cityspaces,默认已经配置好FCN.
1、下载cityspaces
https://www.cityscapes-dataset.com/
需要注册一个账号
cityspaces包含2917个训练样本,与500个测试样本。为了尽量少改源码,将cityspaces训练的原始样本放入到img文件夹下,标注样本放入到cls文件夹下。
建立一个train文件夹,在train文件夹下放入img文件夹,cls文件夹,以及train.txt。
将cityspaces测试的原始样本放入到JPEGImages文件夹下,标注的样本放入到SegmentationClass文件夹下,val.txt放入到ImageSets/Segmentation文件中。
建立一个test文件夹,在test文件夹下放入JPEGImages、SegmentationClass以及ImageSets三个文件夹。
这样测试样本就建立好了。值得注意的是train.txt以及val.txt每一行存储的是不带路径与后缀的图片名。标注文件名与原始图像名称要对应。
2、修改voc_layers.py
FCN训练默认使用的标注样本是mat文件,这里改为标注的png文件。
修改voc_layers.py中的登录标注文件函数load_label。
原始函数如下:
def load_label(self, idx):
"""
Load label image

本文详细介绍了如何使用Cityscapes数据集准备和调整全卷积网络(FCN)进行训练的过程,包括数据集下载、路径修改、标注文件处理及解决常见错误的方法。

5209

被折叠的 条评论
为什么被折叠?



