米虫 Ian
2018-12-25 16:58deeplab v3+训练loss不收敛问题
- 我使用的是官网的代码https://github.com/tensorflow/models/tree/master/research/deeplab 复现deeplab v3+;
- 训练数据就是标准的Pascal voc2012。训练之前已经按照官网上的说法,通过运行脚本download_and_convert_voc2012.sh下载voc2012数据、并将label转换为单通道、并将数据转换为需要的tfrecord格式;
- 训练模型也是从提供的model_zoo下载的https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/model_zoo.md;
- 学习率保持默认,即learning rate=0.0001;
- Linux Ubuntu 16.04;TensorFlow1.6.0 installed from Anaconda;CUDA9.0/cudnn7.0.5;GeForce GTX 1080 Ti;
- 具体训练代码是:
python deeplab/train.py \
--logtostderr \
--training_number_of_steps=30000 \
--train_split="train" \
--model_variant="xception_65" \
--atrous_rates=6 \
--atrous_rates=12 \
--atrous_rates=18 \
--output_stride=16 \
--decoder_output_stride=4 \
--train_crop_size=513 \
--train_crop_size=513 \
--train_batch_size=2 \
--dataset="pascal_voc_seg" \
--fine_tune_batch_norm = False \
--tf_initial_checkpoint="{下载的checkpoint路径}/deeplabv3_pascal_train_aug/model.ckpt.index" \
--train_logdir="{要写入路径}/exp/train_on_train_set/train" \
--dataset_dir="{数据集路径}/pascal_voc_seg/tfrecord"
- 点赞
- 回答
- 收藏
- 复制链接分享
1条回答
为你推荐
- vue3.0+ts 在引用swiper/vue时报错
- vue.js
- typescript
- 5个回答
- vuejs+zTree callback没反应,大佬救命
- html5
- vue.js
- 2个回答
- Google oauth2端点不返回用户个人资料信息(名称等)
- it技术
- 互联网问答
- IT行业问题
- 计算机技术
- 编程语言问答
- 2个回答
- 在同一页面上显示条纹模式,而不是在“https://checkout.stripe.com/v3/...”中显示模式
- laravel
- php
- jquery
- 1个回答
- Youtube Data API V3访问订阅者计数
- api
- youtube
- php
- 1个回答
换一换