deeplabv3+ 从零开始训练

本文介绍了如何从零开始使用deeplabv3+进行深度学习语义分割的训练。首先,下载官方的预训练权重并解压。接着,详细说明了train_aug数据集的处理方式,强调了在训练配置中修改train_split为train的重要性。训练过程中,调整了training_batch_size和checkpoint_dir等参数,并通过两次不同的训练策略(一次允许BN参数微调,一次固定BN参数)来优化模型。作者发现train_aug数据集并未提供最佳结果,可能是由于缺乏背景区域的白边导致。最后,提供了训练脚本和参数介绍的相关链接。
摘要由CSDN通过智能技术生成

首先需要下载官方提供的imagenet以及coco的预训练权重:

http://download.tensorflow.org/models/xception_65_coco_pretrained_2018_10_02.tar.gz

然后将其解压到一个文件夹比如:

 .../deeplab/datasets/pascal_voc_seg/coco_pretrain

train_aug的处理:https://blog.csdn.net/u013249853/article/details/100136780,只需要下载一分钟,仅下载标注,一定要看清楚。

 

 

然后我直接放我的训练配置,注意,python代码是一句没改的。一定要将train_split改成train。否则会将val也一并加入训练。比如下面的保存成train.sh

#!/bin/bash
# Copyright 2018 The TensorFlow Authors All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language gove
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值