Merge pull request #1093 from CellScope/io-cant-load-error-msg

[fix] Move file reading error checking closer to actual file read command
BVLC · Sep 16, 2014 · a77ca76 · a77ca76
2 parents 0fb2faf + aecab61
commit a77ca76
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/caffe/util/io.cpp b/src/caffe/util/io.cpp
@@ -71,16 +71,18 @@ bool ReadImageToDatum(const string& filename, const int label,
   cv::Mat cv_img;
   int cv_read_flag = (is_color ? CV_LOAD_IMAGE_COLOR :
     CV_LOAD_IMAGE_GRAYSCALE);
-  if (height > 0 && width > 0) {
-    cv::Mat cv_img_origin = cv::imread(filename, cv_read_flag);
-    cv::resize(cv_img_origin, cv_img, cv::Size(width, height));
-  } else {
-    cv_img = cv::imread(filename, cv_read_flag);
-  }
-  if (!cv_img.data) {
+
+  cv::Mat cv_img_origin = cv::imread(filename, cv_read_flag);
+  if (!cv_img_origin.data) {
     LOG(ERROR) << "Could not open or find file " << filename;
     return false;
   }
+  if (height > 0 && width > 0) {
+    cv::resize(cv_img_origin, cv_img, cv::Size(height, width));
+  } else {
+    cv_img = cv_img_origin;
+  }
+
   int num_channels = (is_color ? 3 : 1);
   datum->set_channels(num_channels);
   datum->set_height(cv_img.rows);