Display total num of processed files after computing image mean

diff --git a/tools/compute_image_mean.cpp b/tools/compute_image_mean.cpp
index 3e6b911..d1739da 100644 (file)
--- a/tools/compute_image_mean.cpp
+++ b/tools/compute_image_mean.cpp
@@ -66,6 +66,9 @@ int main(int argc, char** argv) {
       LOG(ERROR) << "Processed " << count << " files.";
     }
   }
+  if (count % 10000 != 0) {
+    LOG(ERROR) << "Processed " << count << " files.";
+  }
   for (int i = 0; i < sum_blob.data_size(); ++i) {
     sum_blob.set_data(i, sum_blob.data(i) / count);
   }