enhancement for deepface home path creation

This commit is contained in:
miawoltn 2023-06-11 07:38:27 +01:00
parent 3bf8bcb349
commit 011a0987a7

View File

@ -36,13 +36,15 @@ def initialize_folder():
OSError: if the folder cannot be created. OSError: if the folder cannot be created.
""" """
home = get_deepface_home() home = get_deepface_home()
deepFaceHomePath = home + "/.deepface"
weightsPath = deepFaceHomePath + "/weights"
if not os.path.exists(home + "/.deepface"): if not os.path.exists(deepFaceHomePath):
os.makedirs(home + "/.deepface") os.makedirs(deepFaceHomePath, exist_ok=True)
print("Directory ", home, "/.deepface created") print("Directory ", home, "/.deepface created")
if not os.path.exists(home + "/.deepface/weights"): if not os.path.exists(weightsPath):
os.makedirs(home + "/.deepface/weights") os.makedirs(weightsPath, exist_ok=True)
print("Directory ", home, "/.deepface/weights created") print("Directory ", home, "/.deepface/weights created")