diff --git a/deploy/cpp/src/transforms.cpp b/deploy/cpp/src/transforms.cpp index 626b2053d2473bcf66fcb1a760d9ce2e101324f4..f623fc664e9d66002e0eb0065d034d90965eddf7 100644 --- a/deploy/cpp/src/transforms.cpp +++ b/deploy/cpp/src/transforms.cpp @@ -15,6 +15,7 @@ #include #include #include +#include #include "include/paddlex/transforms.h" @@ -60,8 +61,8 @@ bool ResizeByShort::Run(cv::Mat* im, ImageBlob* data) { data->reshape_order_.push_back("resize"); float scale = GenerateScale(*im); - int width = static_cast(scale * im->cols); - int height = static_cast(scale * im->rows); + int width = static_cast(round(scale * im->cols)); + int height = static_cast(round(scale * im->rows)); cv::resize(*im, *im, cv::Size(width, height), 0, 0, cv::INTER_LINEAR); data->new_im_size_[0] = im->rows;