Merge pull request #142 from zestloveheart/patch-1

Update create_lmdb.py Former-commit-id: 78d4dc7 Former-commit-id: 39b6de4
XPixelGroup · May 28, 2019 · 9bbc011 · 9bbc011
2 parents 3a10e62 + 047b546
commit 9bbc011
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/codes/scripts/create_lmdb.py b/codes/scripts/create_lmdb.py
@@ -22,7 +22,7 @@
     data_size = sum([img.nbytes for img in dataset])
 # mode 2 large data, read imgs later
 elif mode == 2:
-    data_size = sum(os.stat(v).stat().st_size for v in img_list)
+    data_size = sum(os.stat(v).st_size for v in img_list)
 else:
     raise ValueError('mode should be 1 or 2')