diff --git a/csrc/mmdeploy/codebase/mmdet/instance_segmentation.cpp b/csrc/mmdeploy/codebase/mmdet/instance_segmentation.cpp index 3168b7fe8b..34ffdf691c 100644 --- a/csrc/mmdeploy/codebase/mmdet/instance_segmentation.cpp +++ b/csrc/mmdeploy/codebase/mmdet/instance_segmentation.cpp @@ -21,6 +21,7 @@ class ResizeInstanceMask : public ResizeBBox { } operation::Context ctx(device_, stream_); warp_affine_ = operation::Managed::Create("bilinear"); + permute_ = operation::Managed<::mmdeploy::operation::Permute>::Create(); } // TODO: remove duplication @@ -137,7 +138,6 @@ class ResizeInstanceMask : public ResizeBBox { auto mask_height = (int)d_mask.shape(1); auto mask_width = (int)d_mask.shape(2); // (C, H, W) -> (H, W, C) - permute_ = ::mmdeploy::operation::Managed<::mmdeploy::operation::Permute>::Create(); std::vector axes = {1, 2, 0}; OUTCOME_TRY(permute_.Apply(d_mask, d_mask, axes)); Device host{"cpu"};