From 357b484d37d95bc1dbd56af8f5f4568dd9f935d4 Mon Sep 17 00:00:00 2001 From: Yuchen Fan Date: Thu, 7 Jul 2022 00:32:25 -0700 Subject: [PATCH] [Fix] Fix undefined variable error in pixel_group.cpp (#2104) --- mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp b/mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp index 1d185f9456..db06a224a0 100755 --- a/mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp +++ b/mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp @@ -41,7 +41,7 @@ std::vector> pixel_group_cpu( Tensor kernel_contour, int kernel_region_num, float dis_threshold) { assert(score.dim() == 2); assert(mask.dim() == 2); - assert(embedding_dim.dim() == 3); + assert(embedding.dim() == 3); int height = score.size(0); int width = score.size(1); assert(height == mask.size(0) == embedding.size(1) == kernel_label.size(1));