CV:Bugfix: Fix bug for filter

alibaba · Sep 12, 2024 · 7940973 · 7940973
1 parent 1effb0c
commit 7940973
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/cv/source/imgproc/filter.cpp b/tools/cv/source/imgproc/filter.cpp
@@ -46,6 +46,9 @@ static VARP formatOutput(VARP src, halide_type_t type) {
     if (channel == 1) {
         squeeze_dims.push_back(-1);
     }
+    if (!squeeze_dims.empty()) {
+        src = _Squeeze(src, squeeze_dims);
+    }
     if (type == halide_type_of<uint8_t>()) {
         src = _Minimum(src, _Scalar<float>(255));
         src = _Maximum(src, _Scalar<float>(0));