DeepLab 源码分析之 input_preprocess.py

最新推荐文章于 2023-12-20 17:21:50 发布

木材的才

最新推荐文章于 2023-12-20 17:21:50 发布

阅读量1.1k

点赞数

分类专栏： DeepLab Segmentation 文章标签： DeepLab Segmentation

本文链接：https://blog.csdn.net/lifengcai_/article/details/80270955

版权

这次我们分析 input_preprocess.py 主要是预处理数据用于DeepLab训练或验证

使用了 core/preprocess_utils.py 的大量函数

首先 import 必要的库

import tensorflow as tf
from deeplab.core import feature_extractor
from deeplab.core import preprocess_utils

# 训练时左右反转的概率
_PROB_OF_FLIP = 0.5

函数 preprocess_image_and_label
返回原图
处理后的图片 [crop_height, crop_width, 3]
标签 [crop_height, crop_width, 1]

def preprocess_image_and_label(image,
                               label,
                               crop_height,
                               crop_width,
                               min_resize_value=None,
                               max_resize_value=None,
                               resize_factor=None,
                               min_scale_factor=1.,
                               max_scale_factor=1.,
                               scale_factor_step_size=0,
                               ignore_label=