def _initialize(self, use_gpu=False): # 设置模型路径 self.model_path = os.path.join(self.directory, "hand_pose_localization") # 加载模型 self.model = Model(self.model_path, use_gpu)
def __init__(self, name=None, use_gpu=False): # 设置模型路径 self.model_path = os.path.join(self.directory, "hand_pose_localization") # 加载模型 self.model = Model(modelpath=self.model_path, use_gpu=use_gpu, use_mkldnn=False, combined=True)
class Hand_Pose_Localization(Module): # 初始化函数 def __init__(self, name=None, use_gpu=False): # 设置模型路径 self.model_path = os.path.join(self.directory, "hand_pose_localization") # 加载模型 self.model = Model(modelpath=self.model_path, use_gpu=use_gpu, use_mkldnn=False, combined=True) # 关键点检测函数 def keypoint_detection(self, images=None, paths=None, batch_size=1, output_dir='output', visualization=False): # 加载数据处理器 processor = Processor(images, paths, batch_size, output_dir) # 模型预测 outputs = self.model.predict(processor.input_datas) # 结果后处理 results = processor.postprocess(outputs, visualization) # 返回结果 return results # Hub Serving @serving def serving_method(self, images, **kwargs): # 获取输入数据 images_decode = [base64_to_cv2(image) for image in images] # 关键点检测 results = self.keypoint_detection(images_decode, **kwargs) # 返回结果 return results