Merge pull request #960 from littletomatodonkey/sta/fix_cpp_32

fix cpp inference bug

Merge pull request #960 from littletomatodonkey/sta/fix_cpp_32
fix cpp inference bug
d8ed88da · Double_V · GitHub · 54cab7b1 · 5760f73b · d8ed88da
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

deploy/cpp_infer/src/preprocess_op.cpp deploy/cpp_infer/src/preprocess_op.cpp +2 -2

未找到文件。
--- a/deploy/cpp_infer/src/preprocess_op.cpp
+++ b/deploy/cpp_infer/src/preprocess_op.cpp
@@ -85,7 +85,7 @@ void ResizeImgType0::Run(const cv::Mat &img, cv::Mat &resize_img,

  if (resize_w % 32 == 0)
    resize_w = resize_w;
-  else if (resize_w / 32 < 1)
+  else if (resize_w / 32 < 1 + 1e-5)
    resize_w = 32;
  else
    resize_w = (resize_w / 32 - 1) * 32;