From be2df844221476a2f8ee5c1bb03258eadecac047 Mon Sep 17 00:00:00 2001 From: Kevin James Matzen Date: Sun, 12 Oct 2014 14:47:04 -0400 Subject: [PATCH] Added a couple of sanity checks to make sure the datum buffer sizes matched what we expected. --- tools/compute_image_mean.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tools/compute_image_mean.cpp b/tools/compute_image_mean.cpp index 01e16c1..aaa324a 100644 --- a/tools/compute_image_mean.cpp +++ b/tools/compute_image_mean.cpp @@ -62,10 +62,12 @@ int main(int argc, char** argv) { CHECK_EQ(size_in_datum, data_size) << "Incorrect data field size " << size_in_datum; if (data.size() != 0) { + CHECK_EQ(data.size(), size_in_datum); for (int i = 0; i < size_in_datum; ++i) { sum_blob.set_data(i, sum_blob.data(i) + (uint8_t)data[i]); } } else { + CHECK_EQ(datum.float_data_size(), size_in_datum); for (int i = 0; i < size_in_datum; ++i) { sum_blob.set_data(i, sum_blob.data(i) + static_cast(datum.float_data(i))); -- 2.7.4