diff --git a/modules/image/face_detection/pyramidbox_lite_mobile/README.md b/modules/image/face_detection/pyramidbox_lite_mobile/README.md index b4fd8b8c865bfaf14c2b3859a6780044a9180e76..e4f99608afa6ab5da46abd333eb1b8be0c5217d7 100644 --- a/modules/image/face_detection/pyramidbox_lite_mobile/README.md +++ b/modules/image/face_detection/pyramidbox_lite_mobile/README.md @@ -101,19 +101,13 @@ - ```python - def save_inference_model(dirname, - model_filename=None, - params_filename=None, - combined=True) + def save_inference_model(dirname) ``` - 将模型保存到指定路径。 - **参数** - - dirname: 存在模型的目录名称；
- - model\_filename: 模型文件名称，默认为\_\_model\_\_；
- - params\_filename: 参数文件名称，默认为\_\_params\_\_(仅当`combined`为True时生效)；
- - combined: 是否将参数保存到统一的一个文件中。 + - dirname: 模型保存路径
## 四、服务部署 @@ -167,6 +161,10 @@ 移除 fluid api +* 1.3.0 + + 修复无法导出推理模型的问题 + - ```shell - $ hub install pyramidbox_lite_mobile==1.2.1 + $ hub install pyramidbox_lite_mobile==1.3.0 ``` diff --git a/modules/image/face_detection/pyramidbox_lite_mobile/README_en.md b/modules/image/face_detection/pyramidbox_lite_mobile/README_en.md index d1439fc22b9ddfce3d01ce5e9d7aa1cedcf57b94..3c50825efe2039f9a45809c23ec9dc90e7ebc74c 100644 --- a/modules/image/face_detection/pyramidbox_lite_mobile/README_en.md +++ b/modules/image/face_detection/pyramidbox_lite_mobile/README_en.md @@ -100,19 +100,13 @@ - ```python - def save_inference_model(dirname, - model_filename=None, - params_filename=None, - combined=True) + def save_inference_model(dirname) ``` - Save model to specific path - **Parameters** - - dirname: output dir for saving model - - model\_filename: filename for saving model - - params\_filename: filename for saving parameters - - combined: whether save parameters into one file + - dirname: model save path ## IV.Server Deployment @@ -166,6 +160,10 @@ Remove fluid api +* 1.3.0 + + Fix a bug of save_inference_model + - ```shell - $ hub install pyramidbox_lite_mobile==1.2.1 + $ hub install pyramidbox_lite_mobile==1.3.0 ``` diff --git a/modules/image/face_detection/pyramidbox_lite_mobile/module.py b/modules/image/face_detection/pyramidbox_lite_mobile/module.py index 2f6e665e9255e39535aa22255c37d24e928cee5f..2a550bb5fabd79e81e707f338146fbc8f10bd538 100644 --- a/modules/image/face_detection/pyramidbox_lite_mobile/module.py +++ b/modules/image/face_detection/pyramidbox_lite_mobile/module.py @@ -10,11 +10,10 @@ import numpy as np import paddle from paddle.inference import Config from paddle.inference import create_predictor -from pyramidbox_lite_mobile.data_feed import reader -from pyramidbox_lite_mobile.processor import base64_to_cv2 -from pyramidbox_lite_mobile.processor import postprocess +from .data_feed import reader +from .processor import base64_to_cv2 +from .processor import postprocess -import paddlehub as hub from paddlehub.module.module import moduleinfo from paddlehub.module.module import runnable from paddlehub.module.module import serving @@ -25,11 +24,10 @@ from paddlehub.module.module import serving author="baidu-vis", author_email="", summary="PyramidBox-Lite-Mobile is a high-performance face detection model.", - version="1.2.1") -class PyramidBoxLiteMobile(hub.Module): - - def _initialize(self): - self.default_pretrained_model_path = os.path.join(self.directory, "pyramidbox_lite_mobile_face_detection") + version="1.3.0") +class PyramidBoxLiteMobile: + def __init__(self): + self.default_pretrained_model_path = os.path.join(self.directory, "pyramidbox_lite_mobile_face_detection", "model") self._set_config() self.processor = self @@ -37,7 +35,9 @@ class PyramidBoxLiteMobile(hub.Module): """ predictor config setting """ - cpu_config = Config(self.default_pretrained_model_path) + model = self.default_pretrained_model_path+'.pdmodel' + params = self.default_pretrained_model_path+'.pdiparams' + cpu_config = Config(model, params) cpu_config.disable_glog_info() cpu_config.disable_gpu() self.cpu_predictor = create_predictor(cpu_config) @@ -49,7 +49,7 @@ class PyramidBoxLiteMobile(hub.Module): except: use_gpu = False if use_gpu: - gpu_config = Config(self.default_pretrained_model_path) + gpu_config = Config(model, params) gpu_config.disable_glog_info() gpu_config.enable_use_gpu(memory_pool_init_size_mb=1000, device_id=0) self.gpu_predictor = create_predictor(gpu_config) @@ -125,26 +125,6 @@ class PyramidBoxLiteMobile(hub.Module): res.append(out) return res - def save_inference_model(self, dirname, model_filename=None, params_filename=None, combined=True): - if combined: - model_filename = "__model__" if not model_filename else model_filename - params_filename = "__params__" if not params_filename else params_filename - place = paddle.CPUPlace() - exe = paddle.Executor(place) - - program, feeded_var_names, target_vars = paddle.static.load_inference_model( - dirname=self.default_pretrained_model_path, executor=exe) - - var = program.global_block().vars['detection_output_0.tmp_1'] - - paddle.static.save_inference_model(dirname=dirname, - main_program=program, - executor=exe, - feeded_var_names=feeded_var_names, - target_vars=target_vars, - model_filename=model_filename, - params_filename=params_filename) - @serving def serving_method(self, images, **kwargs): """ diff --git a/modules/image/face_detection/pyramidbox_lite_mobile/processor.py b/modules/image/face_detection/pyramidbox_lite_mobile/processor.py index 5057ab5b1fc6c84fdd06a7fcc45336031fa2ba4d..2045f51c21be5a437c196ca86a41b6c2cf61d64d 100644 --- a/modules/image/face_detection/pyramidbox_lite_mobile/processor.py +++ b/modules/image/face_detection/pyramidbox_lite_mobile/processor.py @@ -5,7 +5,6 @@ from __future__ import print_function import os import time -from collections import OrderedDict import base64 import cv2 diff --git a/modules/image/face_detection/pyramidbox_lite_mobile/test.py b/modules/image/face_detection/pyramidbox_lite_mobile/test.py new file mode 100644 index 0000000000000000000000000000000000000000..1b06f9b7f23dd0063a62fb5f5c1a60ce33adbe9f --- /dev/null +++ b/modules/image/face_detection/pyramidbox_lite_mobile/test.py @@ -0,0 +1,133 @@ +import os +import shutil +import unittest + +import cv2 +import requests +import paddlehub as hub + + +os.environ['CUDA_VISIBLE_DEVICES'] = '0' + + +class TestHubModule(unittest.TestCase): + @classmethod + def setUpClass(cls) -> None: + img_url = 'https://ai-studio-static-online.cdn.bcebos.com/7799a8ccc5f6471b9d56fb6eff94f82a08b70ca2c7594d3f99877e366c0a2619' + if not os.path.exists('tests'): + os.makedirs('tests') + response = requests.get(img_url) + assert response.status_code == 200, 'Network Error.' + with open('tests/test.jpg', 'wb') as f: + f.write(response.content) + cls.module = hub.Module(name="pyramidbox_lite_mobile") + + @classmethod + def tearDownClass(cls) -> None: + shutil.rmtree('tests') + shutil.rmtree('inference') + shutil.rmtree('detection_result') + + def test_face_detection1(self): + results = self.module.face_detection( + paths=['tests/test.jpg'], + use_gpu=False, + visualization=False + ) + bbox = results[0]['data'][0] + + confidence = bbox['confidence'] + left = bbox['left'] + right = bbox['right'] + top = bbox['top'] + bottom = bbox['bottom'] + + self.assertTrue(confidence > 0.5) + self.assertTrue(1000 < left < 4000) + self.assertTrue(1000 < right < 4000) + self.assertTrue(0 < top < 2000) + self.assertTrue(0 < bottom < 2000) + + def test_face_detection2(self): + results = self.module.face_detection( + images=[cv2.imread('tests/test.jpg')], + use_gpu=False, + visualization=False + ) + bbox = results[0]['data'][0] + + confidence = bbox['confidence'] + left = bbox['left'] + right = bbox['right'] + top = bbox['top'] + bottom = bbox['bottom'] + + self.assertTrue(confidence > 0.5) + self.assertTrue(1000 < left < 4000) + self.assertTrue(1000 < right < 4000) + self.assertTrue(0 < top < 2000) + self.assertTrue(0 < bottom < 2000) + + def test_face_detection3(self): + results = self.module.face_detection( + images=[cv2.imread('tests/test.jpg')], + use_gpu=False, + visualization=True + ) + bbox = results[0]['data'][0] + + confidence = bbox['confidence'] + left = bbox['left'] + right = bbox['right'] + top = bbox['top'] + bottom = bbox['bottom'] + + self.assertTrue(confidence > 0.5) + self.assertTrue(1000 < left < 4000) + self.assertTrue(1000 < right < 4000) + self.assertTrue(0 < top < 2000) + self.assertTrue(0 < bottom < 2000) + + def test_face_detection4(self): + results = self.module.face_detection( + images=[cv2.imread('tests/test.jpg')], + use_gpu=True, + visualization=False + ) + bbox = results[0]['data'][0] + + confidence = bbox['confidence'] + left = bbox['left'] + right = bbox['right'] + top = bbox['top'] + bottom = bbox['bottom'] + + self.assertTrue(confidence > 0.5) + self.assertTrue(1000 < left < 4000) + self.assertTrue(1000 < right < 4000) + self.assertTrue(0 < top < 2000) + self.assertTrue(0 < bottom < 2000) + + def test_face_detection5(self): + self.assertRaises( + AssertionError, + self.module.face_detection, + paths=['no.jpg'] + ) + + def test_face_detection6(self): + self.assertRaises( + AttributeError, + self.module.face_detection, + images=['test.jpg'] + ) + + def test_save_inference_model(self): + self.module.save_inference_model('./inference/model') + + self.assertTrue(os.path.exists('./inference/model.pdmodel')) + self.assertTrue(os.path.exists('./inference/model.pdiparams')) + + +if __name__ == "__main__": + unittest.main()