From: Evan Shelhamer Date: Fri, 6 Feb 2015 00:41:46 +0000 (-0800) Subject: die on inputs to IMAGE_DATA that fail to load X-Git-Tag: submit/tizen/20180823.020014~572^2~32^2 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e80b60fc1e64de7aacd717f70446afdd34ba6845;p=platform%2Fupstream%2Fcaffeonacl.git die on inputs to IMAGE_DATA that fail to load It's better to know than march silently on. --- diff --git a/src/caffe/layers/image_data_layer.cpp b/src/caffe/layers/image_data_layer.cpp index ef6a342..1366948 100644 --- a/src/caffe/layers/image_data_layer.cpp +++ b/src/caffe/layers/image_data_layer.cpp @@ -118,10 +118,8 @@ void ImageDataLayer::InternalThreadEntry() { timer.Start(); CHECK_GT(lines_size, lines_id_); cv::Mat cv_img = ReadImageToCVMat(root_folder + lines_[lines_id_].first, - new_height, new_width, is_color); - if (!cv_img.data) { - continue; - } + new_height, new_width, is_color); + CHECK(cv_img.data) << "Could not load " << lines_[lines_id_].first; read_time += timer.MicroSeconds(); timer.Start(); // Apply transformations (mirror, crop...) to the image