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

fix: yolo input_shape

Browse Source
This commit is contained in:
roberto-corno-nttdata 2024-12-09 16:04:08 +01:00
parent d3a3f2b65f
commit c25f14af10
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/models/facial_recognition/Yolo.py
View File

@ -37,7 +37,7 @@ class YoloFacialRecognitionClient(FacialRecognition):
def __init__(self, model: YoloModel): def __init__(self, model: YoloModel):
super().__init__() super().__init__()
self.model_name = "Yolo" self.model_name = "Yolo"
self.input_shape = (224, 224) self.input_shape = (640, 640)
self.output_shape = 512 self.output_shape = 512
self.model = self.build_model(model) self.model = self.build_model(model)