微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

python-Tensorflow:裁剪图像的最大中央正方形区域

我的网络拍摄的图像尺寸为100 x 100像素.因此,我必须调整数据集大小不同的图像的大小.我希望能够从给定图像中提取最大的中央正方形区域,然后将其调整为100 x 100.

更准确地说,假设图像的宽度为200像素,高度为50像素.然后,我想提取最大的中央正方形区域,在此示例中为50 x 50,然后将图像调整为100 x 100像素.

使用Tensorflow的正确方法是什么?现在我正在使用tf.image.resize_images()扭曲图像,我想摆脱它.

解决方法:

听起来crop_to_bounding_box可以满足您的需求:

import tensorflow as tf

def crop_center(image):
    h, w = image.shape[-3], image.shape[-2]
    if h > w:
        cropped_image = tf.image.crop_to_bounding_Box(image, (h - w) // 2, 0, w, w)
    else:
        cropped_image = tf.image.crop_to_bounding_Box(image, 0, (w - h) // 2, h, h)
    return tf.image.resize_images(cropped_image, (100, 100))

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐