[Fix] Fix undefined variable error in pixel_group.cpp (#2104)

open-mmlab · Jul 7, 2022 · 357b484 · 357b484
1 parent d77557b
commit 357b484
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp b/mmcv/ops/csrc/pytorch/cpu/pixel_group.cpp
@@ -41,7 +41,7 @@ std::vector<std::vector<float>> 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));