如何改善DeeplabV3 +分割边界/边缘?

问题描述

我正在使用标准研究模型在mobilenet_v3_large_seg顶部的Tensorflow 1.15.3中训练自定义数据集。一般而言,训练尚可,但是边界/边缘非常嘈杂,什么可以帮助改善边界更好的检测段?

python deeplab/train.py --logtostderr --training_number_of_steps=84000 \
--output_stride 4 \
--train_split="trainval" --model_variant="mobilenet_v3_large_seg" \
--train_crop_size="384,384" --dataset="123" --save_interval_secs=300 \
--save_summaries_secs=300 --save_summaries_images=True --log_steps=100 --train_logdir=123/training \
--dataset_dir=123 --min_scale_factor=0.75 --max_scale_factor=1.25 \
--fine_tune_batch_norm=True --train_batch_size=4 --scale_factor_step_size=0.05

解决方法

通过进一步的测试和调整,有两点改善了我的结果:

  1. 嘈杂的图像很难进行分割,因此我之前使用均值/中值滤波器对图像进行了平滑处理。

  2. 培训确实需要大量的样本和时间,并且会慢慢改善。