add if isfile() to get_hash()

5 years ago · 59beae98e8
parent 6649770001
commit 59beae98e8
1 changed files with 1 additions and 1 deletions
--- a/utils/datasets.py
+++ b/utils/datasets.py
@ -28,7 +28,7 @@ for orientation in ExifTags.TAGS.keys():

 def get_hash(files):
    # Returns a single hash value of a list of files
-    return sum(os.path.getsize(f) for f in files)
+    return sum(os.path.getsize(f) for f in files if os.path.isfile(f))


 def exif_size(img):