x_write=x.cpu().numpy().transpose(0,2,3,1)
x_ini=[]
for i in range(x_write.shape[0]):
    x_aux=x_write[i]
    # x_aux_int=x_aux.astype(int)
    x_aux_r=cv2.normalize(x_aux, None, 0, 255, cv2.NORM_MINMAX).astype(int)
    cv2.imwrite('x_aux_r'+str(i)+'.png',x_aux_r)
Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐