FCN---训练cityspaces

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

      未经允许,不要随意转载!

     最近想试试用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文件夹下放入JPEGImagesSegmentationClass以及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

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值