update

infer_api.py

@@ -200,6 +200,7 @@ def inference(validation_pipeline, input_image, vae, feature_extractor, image_en
     print("OUT!!!!!!")
 
     img_buf = io.BytesIO()
+    save_image(out[0], img_buf, format='PNG')
     img_buf.seek(0)
     img = Image.open(img_buf)
 
@@ -254,8 +255,11 @@ class TestConfig:
 def convert_to_numpy(tensor):
     return tensor.mul(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to("cpu", torch.uint8).numpy()
 
-def save_image(tensor):
+# 定义一个函数，用于保存图像
+def save_image2(tensor):
+    # 将tensor转换为numpy数组
     ndarr = convert_to_numpy(tensor)
+    # 调用save_image_numpy函数，保存图像
     return save_image_numpy(ndarr)
 
 def save_image_numpy(ndarr):
@@ -324,8 +328,8 @@ def run_multiview_infer(data, pipeline, cfg: TestConfig, num_levels=3):
                 color = images_pred[idx]
 
                 ## save color and normal---------------------
-                new_normal = save_image(normal)
-                new_color = save_image(color)
+                new_normal = save_image2(normal)
+                new_color = save_image2(color)
 
                 if num_levels == 2:
                     results[level+1]['normals'].append(new_normal)