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

added weights

Browse Source
This commit is contained in:
Raghucharan16 2025-02-20 23:25:36 +05:30
parent 5a73d91744
commit ad0c7e75e7
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
deepface/models/facial_recognition/Buffalo_L.py
View File

@ -2,6 +2,9 @@ import cv2
import numpy as np
from deepface.models.FacialRecognition import FacialRecognition
from deepface.commons.logger import Logger
from deepface.basemodel import get_weights_path
from deepface.common import weight_utils
import os
logger = Logger()
@ -22,11 +25,16 @@ class Buffalo_L(FacialRecognition):
self.load_model()
def load_model(self):
"""
Load the InsightFace Buffalo_L recognition model.
"""
self.model = get_model('buffalo_l/w600k_r50.onnx', download=True)
self.model.prepare(ctx_id=-1, input_size=self.input_shape) # ctx_id=-1 for CPU
root = os.path.join(get_weights_path(), 'insightface')
model_name = 'buffalo_l/w600k_r50.onnx'
model_path = os.path.join(root, model_name)
if not os.path.exists(model_path):
url = 'https://drive.google.com/file/d/1N0GL-8ehw_bz2eZQWz2b0A5XBdXdxZhg/view?usp=sharing'
weight_utils.download_file(url, model_path)
self.model = get_model(model_name, root=root)
self.model.prepare(ctx_id=-1, input_size=self.input_shape)
def preprocess(self, img):
"""