首页 > 其他分享 >pytorch中使用vutils对多张图像进行拼接 (import torchvision.utils as vutils)

pytorch中使用vutils对多张图像进行拼接 (import torchvision.utils as vutils)

时间:2022-09-23 10:46:23浏览次数:55  
标签:tensor torchvision utils vutils images import png 1000

1.png

 

 

 

2.png

 

 

 

 

 

在pytorch中使用torchvision的vutils函数实现对多张图片的拼接。具体操作就是将上面的两张图片,1.png和2.png的多张图片进行拼接形成一张图片,拼接后的效果如下图。

 

 

 

 

给出具体代码:

import matplotlib.pyplot as plt
from PIL import Image
import numpy as np
import torch
import torchvision.utils as vutils

im1=Image.open("1.png").convert("RGB")
im1 = im1.resize((1000, 1000)).rotate(-90)

im2=Image.open("2.png").convert("RGB")
im2 = im2.resize((1000, 1000)).rotate(-90)


# 1000, 1000, 3 => 3, 1000, 1000
images = [np.moveaxis(np.array(im1), 2, 0), np.moveaxis(np.array(im2), 2, 0)]*8


images_tensor = vutils.make_grid(torch.tensor(images)/255.0, nrow=4, padding=0, normalize=True)
print(images_tensor.shape)
# 3, 1000, 1000 => 1000, 1000, 3 
plt.imshow(images_tensor.numpy().transpose((1,2,0)))
plt.show()

vutils.save_image(images_tensor, "3.png")
vutils.save_image(images_tensor, "3_back.png", nrow=2, padding=0, normalize=True)
vutils.save_image(torch.tensor(images)/255.0, "4.png", nrow=8, padding=0, normalize=True)

 

 

 

=============================================

 

标签:tensor,torchvision,utils,vutils,images,import,png,1000
From: https://www.cnblogs.com/devilmaycry812839668/p/16721875.html

相关文章

  • Bean复制的几种框架性能比较(Apache BeanUtils、PropertyUtils,Spring BeanUtils,Cglib
    引用于:https://www.cnblogs.com/kaka/archive/2013/03/06/2945514.html     比较的是四种复制的方式,分别为Apache的BeanUtils和PropertyUtils,Spring的BeanUtils......
  • windows系统 python3.6(Anaconda3)安装对应版本 torch、torchvision
    一、官网下载.whl文件https://download.pytorch.org/whl/torch_stable.html  二、使用pip命令安装打开你的anaconda,选择对应虚拟环境终端,进入刚才下载.whl 文件......
  • FileUtils工具类
    导入第三方jar:commons-io@Testpublicvoidtest7()throwsIOException{FilesrcFile=newFile("hello.txt");FiledestFile=newFile("hel......
  • 关于BeanUtils.populate()方法的疑问?
    前几天的在写前端register.jsp的时候,发现前端需要封装的对象好多,网上查了下,竟然可以用BeanUtils来封装,于是在RegistUserServlet中用了BeanUtils来封装成user对象。......
  • 【Java】FileUtils-获取路径的所有文件(或文件夹)
    一、获取指定路径下的所有Excel文件packagecom.boulderaitech.utils;importjava.io.File;importjava.util.Arrays;publicclassFileUtil{publicstaticvo......
  • ReflectUtils
    packagecom.denkou.project.service.zxzy.utils;importcom.denkou.project.service.zxzy.exception.ApiException;importio.swagger.annotations.ApiModelProperty;......
  • juypter notebook中报找不到scipy,torchvision的问题
    在初入深度学习使用juypter这块经常遇到各种问题,每次都被搞的很痛苦; 下面给大家带来我的一点问题解决方案: 首先检查下anaconda中有没有安装scipy这些模块,没有的话在......
  • torch.utils.data
    classtorch.utils.data.Dataset表示\(Dataset\)的抽象类。所有其他数据集都应该进行子类化。所以子类应该覆写__len__和__getitem__,前者提供了数据集的大小,后者支持......
  • Java的HexUtils
    自己写的一个HexUtils,记录下好Copy。1.pom相关 <alg-bc.version>1.71</alg-bc.version><dependency><groupId>org.bouncycastle</groupId>......
  • keras.utils.to_categorical方法
    用法:to_categorical(y,num_classes=None,dtype='float32')将整型的类别标签转为onehot编码。y为int数组,num_classes为标签类别总数,大于max(y)(标签从0开始的)。返回:如果n......