如何解决如何使用 torchvision.io.read_image 将图像作为变量而不是存储文件?
torchvision.io.read_image
用作存储在 path
参数中的输入文件。如果图像存储为变量,如何实现相同的输出?当然,我可以将图像保存为文件,然后从中读取,但这需要额外的时间。有没有办法以输入作为变量而不是路径获得与 torchvision.io.read_image
相同的结果?
解决方法
如果内存中的图像是 PIL 图像,可以使用变换函数将其转换为正确格式的张量(实现与 torchvision.io.read_image
相同的效果,而无需从磁盘中读取内容)。
import PIL
import torchvision.transforms.functional as transform
# Reads a file using pillow
PIL_image = PIL.Image.open(image_path)
# The image can be converted to tensor using
tensor_image = transform.to_tensor(PIL_image)
# The tensor can be converted back to PIL using
new_PIL_image = transform.to_pil_image(tensor_image)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。