Yolov7Face add predict's python Interface (#1039)

* fit yolov7face file path

* TODO:添加yolov7facePython接口Predict

* resolve yolov7face.py

* resolve yolov7face.py
This commit is contained in:
CoolCola
2023-01-04 13:25:34 +08:00
committed by GitHub
parent 406db67a15
commit 3734639226

View File

@@ -140,6 +140,13 @@ class YOLOv7Face(FastDeployModel):
assert self.initialized, "YOLOv7Face initialize failed."
def predict(self, input_image):
"""Detect the location and key points of human faces from an input image
:param input_image: (numpy.ndarray)The input image data, 3-D array with layout HWC, BGR format
:return: FaceDetectionResult
"""
return self._model.predict(input_image)
def batch_predict(self, images):
"""Classify a batch of input image