写入图像的时候有一些常用的方法可供选择,在用 torch 保存的时候,发现tochvision.utils.save_image
的时间有点长,做了个简单的对比:
import cv2
from torchvision.utils import save_image
import torch
import numpy as np
import time
import os
from imageio import imwrite
folder = './.TESTUSE'
img = np.ones((3, 2336, 3504), np.uint8)
# cv2
time1 = time.time()
save_name = os.path.join(folder, 'cv2.png')
cv2.imwrite(save_name, img)