When writing an image, there are some common methods to choose from, when saved with torch, found that the tochvision.utils.save_image time is a bit long, made a simple comparison:
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)
time.time ()-time1
# 0.0007767677307128906
# ImageIO
time1 = Time.time ()
save_name = os.path.join (folder, ' Imageio.png ')
imwrite (Save_name, Img.transpose (1, 2, 0))
time.time ()-time1
# 0.7609069347381592
# torchvision
time1 = Time.time ()
save_name = os.path.join (folder, ' Torchvision.png ')
save_image (torch. Tensor (IMG), save_name)
time.time ()-time1
# 1.1047229766845703