Unverified Commit 357b484d authored by Yuchen Fan's avatar Yuchen Fan Committed by GitHub
Browse files

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

parent d77557b0
......@@ -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));
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment