fix im2rec resize bug

turi-code · May 16, 2015 · b0671d7 · b0671d7
1 parent a79ef04
commit b0671d7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tools/im2rec.cc b/tools/im2rec.cc
@@ -107,11 +107,11 @@ int main(int argc, char *argv[]) {
       CHECK(img.data != NULL) << "OpenCV decode fail:" << path;
       cv::Mat res;
       if (img.rows > img.cols) {
-        cv::resize(img, res, cv::Size(img.rows * new_size / img.cols, 
-                  new_size), 0, 0, CV_INTER_LINEAR);
+        cv::resize(img, res, cv::Size(new_size, img.rows * new_size / img.cols),
+                0, 0, CV_INTER_LINEAR);
       } else {
-        cv::resize(img, res, cv::Size(new_size, new_size * img.cols
-          / img.rows), 0, 0, CV_INTER_LINEAR);
+        cv::resize(img, res, cv::Size(new_size * img.cols / img.rows, new_size),
+                0, 0, CV_INTER_LINEAR);
       }
       encode_buf.clear();
       CHECK(cv::imencode(".jpg", res, encode_buf, encode_params));