From a2c2f3ce57e818a4db55e98344163410646d9997 Mon Sep 17 00:00:00 2001 From: linmin Date: Fri, 21 Feb 2014 11:25:22 +0800 Subject: [PATCH] move if outside of forloop --- tools/compute_image_mean.cpp | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tools/compute_image_mean.cpp b/tools/compute_image_mean.cpp index 771f5dc..3e6b911 100644 --- a/tools/compute_image_mean.cpp +++ b/tools/compute_image_mean.cpp @@ -52,12 +52,13 @@ int main(int argc, char** argv) { const string& data = datum.data(); size_in_datum = std::max(datum.data().size(), datum.float_data_size()); CHECK_EQ(size_in_datum, data_size) << "Incorrect data field size " << size_in_datum; - << data.size(); - for (int i = 0; i < size_in_datum; ++i) { - if (data.size() != 0) { - sum_blob.set_data(i, sum_blob.data(i) + (uint8_t)data[i]); - } else { - sum_blob.set_data(i, sum_blob.data(i) + (float)datum.float_data(i)); + if (data.size() != 0) { + for (int i = 0; i < size_in_datum; ++i) { + sum_blob.set_data(i, sum_blob.data(i) + (uint8_t)data[i]); + } + } else { + for (int i = 0; i < size_in_datum; ++i) { + sum_blob.set_data(i, sum_blob.data(i) + (float)datum.float_data(i)); } } ++count; -- 2.7.4