ceil for image size

scheckmedia · Jun 4, 2020 · b058744 · b058744
1 parent 6e09695
commit b058744
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/inference.cpp b/src/inference.cpp
@@ -75,8 +75,8 @@ int Inference::prepare_image(std::string &filepath, cv::Mat &dest) {
 
   float scale =
       static_cast<float>(CNN_INPUT_SIZE) / std::min(image.rows, image.cols);
-  float nw = round(image.cols * scale);
-  float nh = round(image.rows * scale);
+  float nw = ceil(image.cols * scale);
+  float nh = ceil(image.rows * scale);
   float half = CNN_INPUT_SIZE / 2.0;
   float cx = nw / 2.0;
   float cy = nh / 2.0;