def tf2(img): tf = tf1(img) tf_center, tf_uncenter = data.build_center_uncenter_transforms(img.shape) tf_rot = data.build_augmentation_transform(rotation=45) tf_rot = tf_uncenter + tf_rot + tf_center return tf + tf_rot