写入图像的时间对比

写入图像的时候有一些常用的方法可供选择，在用 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)