hw3

app.py

@@ -107,6 +107,7 @@ def restore_image(pil_img: Image.Image) -> Image.Image:
         output = normalize(output, -1 * mean / std, 1 / std)
         output = output.data.squeeze().float().cpu().clamp_(0, 1).numpy()  # (3,H,W)
         output = (output * 255.0).round().astype(np.uint8)
+        output = np.transpose(output, (1, 2, 0))
         out_pil = Image.fromarray(output, mode="RGB")
         return out_pil
     except Exception as e: