def batch_imgs_reshape_and_agu(batch_imgs, hps):
"""
将images数据shape由 (batch,hps.img_width*hps.img_width*hps.img_depth) -> (batch, hps.img_width, hps.img_width, hps.img_depth)
并且进行数据增强
:param batch_imgs:
:param hps:
:return:
"""
batch_imgs = imgUtil.batch_imgs_reshape(batch_imgs, hps)
batch_imgs = imgUtil.batch_imgs_aug(batch_imgs)
return batch_imgs
def batch_imgs_process(batch_imgs, hps):
batch_imgs = imgUtil.batch_imgs_reshape(batch_imgs, hps)
batch_imgs = imgUtil.batch_imgs_resize(batch_imgs, hps.des_img_size, hps)
return batch_imgs