offline_stores_try_on / util /image_writer.py
Ali Mohsin
feat: Add virtual try-on system components including DensePose, SMPL, and pix2pixHD models, rendering, and utilities.
5db43ff
import cv2
import numpy as np
def save_image(path:str,img:np.ndarray,isRGB=True):
if isRGB and img.ndim==3:
img=img[:,:,[2,1,0]]
cv2.imwrite(path,img)