Fix cv:resize bug

2025-12-24 13:28:13 +08:00 · 2022-12-05 10:10:27 +00:00
parent fdea6f1454
commit 49559a8484
1 changed files with 2 additions and 1 deletions
--- a/examples/multimodal/stable_diffusion/cpp/pipeline_stable_diffusion_inpaint.cc
+++ b/examples/multimodal/stable_diffusion/cpp/pipeline_stable_diffusion_inpaint.cc
@@ -49,7 +49,8 @@ void StableDiffusionInpaintPipeline::PrepareMaskAndMaskedImage(
      float_mask[i] = 1;
    }
  }
-  image_mask.SetExternalData({1, 1, shape[1] * 8, shape[0] * 8},
+  // NCHW format
+  image_mask.SetExternalData({1, 1, shape[0] * 8, shape[1] * 8},
                             FDDataType::FP32, float_mask.data());

  // Set mask_image