github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

throw meaningful error if yolo weights failed while downloading

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-02-24 15:31:50 +00:00
parent af73d3dfe4
commit ab38a1fcf7
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
deepface/detectors/Yolo.py
View File

@ -43,8 +43,15 @@ class YoloClient(Detector):
# Download the model's weights if they don't exist
if not os.path.isfile(weight_path):
gdown.download(WEIGHT_URL, weight_path, quiet=False)
logger.info(f"Downloaded YOLO model {os.path.basename(weight_path)}")
logger.info(f"Downloading Yolo weights from {WEIGHT_URL} to {weight_path}...")
try:
gdown.download(WEIGHT_URL, weight_path, quiet=False)
except Exception as err:
raise ValueError(
f"Exception while downloading Yolo weights from {WEIGHT_URL}."
f"You may consider to download it to {weight_path} manually."
) from err
logger.info(f"Yolo model is just downloaded to {os.path.basename(weight_path)}")
# Return face_detector
return YOLO(weight_path)