diff --git a/deepface/commons/functions.py b/deepface/commons/functions.py index 81b431e..13751cc 100644 --- a/deepface/commons/functions.py +++ b/deepface/commons/functions.py @@ -7,6 +7,19 @@ from keras.preprocessing import image import cv2 from pathlib import Path import gdown +import hashlib + +def findFileHash(file): + BLOCK_SIZE = 65536 # The size of each read from the file + + file_hash = hashlib.sha256() # Create the hash object, can use something other than `.sha256()` if you wish + with open(file, 'rb') as f: # Open the file to read it's bytes + fb = f.read(BLOCK_SIZE) # Read from the file. Take in the amount declared above + while len(fb) > 0: # While there is still data being read from the file + file_hash.update(fb) # Update the hash + fb = f.read(BLOCK_SIZE) # Read the next block from the file + + return file_hash.hexdigest() def initializeFolder(): @@ -19,6 +32,34 @@ def initializeFolder(): if not os.path.exists(home+"/.deepface/weights"): os.mkdir(home+"/.deepface/weights") print("Directory ",home,"/.deepface/weights created") + + #---------------------------------- + """ + #avoid interrupted file download + + weight_hashes = [ + ['age_model_weights.h5', '0aeff75734bfe794113756d2bfd0ac823d51e9422c8961125b570871d3c2b114'] + , ['facenet_weights.h5', '90659cc97bfda5999120f95d8e122f4d262cca11715a21e59ba024bcce816d5c'] + , ['facial_expression_model_weights.h5', 'e8e8851d3fa05c001b1c27fd8841dfe08d7f82bb786a53ad8776725b7a1e824c'] + , ['gender_model_weights.h5', '45513ce5678549112d25ab85b1926fb65986507d49c674a3d04b2ba70dba2eb5'] + , ['openface_weights.h5', '5b41897ec6dd762cee20575eee54ed4d719a78cb982b2080a87dc14887d88a7a'] + , ['race_model_single_batch.h5', 'eb22b28b1f6dfce65b64040af4e86003a5edccb169a1a338470dde270b6f5e54'] + , ['vgg_face_weights.h5', '759266b9614d0fd5d65b97bf716818b746cc77ab5944c7bffc937c6ba9455d8c'] + ] + + for i in weight_hashes: + + weight_file = home+"/.deepface/weights/"+i[0] + expected_hash = i[1] + + #check file exits + if os.path.isfile(weight_file) == True: + current_hash = findFileHash(weight_file) + if current_hash != expected_hash: + print("hash violated for ", i[0],". It's going to be removed.") + os.remove(weight_file) + """ + #---------------------------------- def findThreshold(model_name, distance_metric): diff --git a/tests/unit_tests.py b/tests/unit_tests.py index 7fb570a..5555212 100644 --- a/tests/unit_tests.py +++ b/tests/unit_tests.py @@ -72,4 +72,4 @@ accuracy = round(accuracy, 2) if accuracy > 80: print("Unit tests are completed successfully. Score: ",accuracy,"%") else: - raise ValueError("Unit test score does not satisfy the minimum required accuracy. Minimum expected score is 80% but this got ",accuracy,"%") + raise ValueError("Unit test score does not satisfy the minimum required accuracy. Minimum expected score is 80% but this got ",accuracy,"%") \ No newline at end of file