From 75607bddec4585ee1e9f6d9936a141e2d43fc590 Mon Sep 17 00:00:00 2001
From: Iridoudou <2534936416@qq.com>
Date: Tue, 10 Aug 2021 10:10:53 +0800
Subject: [PATCH] add file counter

---
 fit/tools/main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/fit/tools/main.py b/fit/tools/main.py
index adda852..e05473f 100644
--- a/fit/tools/main.py
+++ b/fit/tools/main.py
@@ -90,9 +90,11 @@ if __name__ == "__main__":
         gender=cfg.MODEL.GENDER,
         model_root='smplpytorch/native/models')
     
+    file_num = 0
     for root,dirs,files in os.walk(cfg.DATASET.PATH):
         for file in files:
-            logger.info('Processing file: {}'.format(file))
+            file_num += 1
+            logger.info('Processing file: {}   [{} / {}]'.format(file,file_num,len(files)))
             target = torch.from_numpy(transform(args.dataset_name, 
                                                 load(args.dataset_name,
                                                         os.path.join(root,file)))).float()