运行DeepLabv2

DeepLab:带深度卷积网、空洞卷积和全连接CRF的语义图像分割 论文 讲解


测试单张图片

  1. 下载源码,修改源码文件夹名称为DeepLabv2。基于Caffe的源码在
  2. 下载模型,解压缩得到deeplab_resnet.ckpt和deeplab_resnet_init.ckpt,把这两个文件放入DeepLabv2/output文件夹。找一个图片1.jpg也放入该文件夹。
  3. 在DeepLabv2创建1infer.sh:
    python inference.py output/1.jpg output/deeplab_resnet.ckpt
    
    #使用自己训练的结果:训练的输出目录是output/snapshots
    python inference.py output/1.jpg output/snapshots/model.ckpt-20000

     

  4. 运行1infer.sh即生成DeepLabv2/output/mask.png。

训练

  1. 下载数据集,和增广数据集。根据运行错误稍微调整一下文件夹位置。
  2. 修改DeepLabv2/train.py中的路径。注意如果RESTORE_FROM指向下载的'output/deeplab_resnet.ckpt'路径,就是在作者的训练结果上继续训练,这样开始训练就是收敛的。
  3. 执行:python train.py --random-mirror --random-scale。

可视化

  1. 在train.py中设置了:SNAPSHOT_DIR = 'output/snapshots/'。
  2. 在DeepLabv2中打开终端,输入:tensorboard --logdir=output/snapshots,按提示打开链接。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值