diff --git a/hub_module/modules/image/face_detection/pyramidbox_lite_mobile/module.py b/hub_module/modules/image/face_detection/pyramidbox_lite_mobile/module.py index 77d05bc336b7b421b6a3fa9c8343a71c68ed157e..8d0294a55eb59f83011b90dce8e4b8369dc1e066 100644 --- a/hub_module/modules/image/face_detection/pyramidbox_lite_mobile/module.py +++ b/hub_module/modules/image/face_detection/pyramidbox_lite_mobile/module.py @@ -28,6 +28,7 @@ class PyramidBoxLiteMobile(hub.Module): self.default_pretrained_model_path = os.path.join( self.directory, "pyramidbox_lite_mobile_face_detection") self._set_config() + self.processor = self def _set_config(self): """ @@ -130,6 +131,9 @@ class PyramidBoxLiteMobile(hub.Module): program, feeded_var_names, target_vars = fluid.io.load_inference_model( dirname=self.default_pretrained_model_path, executor=exe) + var = program.global_block().vars['detection_output_0.tmp_1'] + var.desc.set_dtype(fluid.core.VarDesc.VarType.INT32) + fluid.io.save_inference_model( dirname=dirname, main_program=program, diff --git a/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/module.py b/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/module.py index 91b1adfada5e331abbdd22be83f0f3043fd50eae..e98c9944ff76c8baa97988580fd994c5431048d7 100644 --- a/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/module.py +++ b/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/module.py @@ -37,6 +37,7 @@ class PyramidBoxLiteMobileMask(hub.Module): else: self.face_detector = face_detector_module self._set_config() + self.processor = self def _set_config(self): """ diff --git a/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/processor.py b/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/processor.py index 3e3bf1515143cfbcd1dccda71b03cd3b0cf1e77b..61c7be6addd3c1791d1c8e46d0bcd58dcf93e8c9 100644 --- a/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/processor.py +++ b/hub_module/modules/image/face_detection/pyramidbox_lite_mobile_mask/processor.py @@ -94,12 +94,12 @@ def draw_bounding_box_on_image(save_im_path, output_data): box_fill = (255) text_fill = (0) - draw.rectangle( - xy=(bbox['left'], bbox['top'] - (textsize_height + 5), - bbox['left'] + textsize_width + 10, bbox['top'] - 3), - fill=box_fill) - draw.text( - xy=(bbox['left'], bbox['top'] - 15), text=text, fill=text_fill) + draw.rectangle( + xy=(bbox['left'], bbox['top'] - (textsize_height + 5), + bbox['left'] + textsize_width + 10, bbox['top'] - 3), + fill=box_fill) + draw.text( + xy=(bbox['left'], bbox['top'] - 15), text=text, fill=text_fill) image.save(save_im_path) diff --git a/hub_module/modules/image/face_detection/pyramidbox_lite_server/module.py b/hub_module/modules/image/face_detection/pyramidbox_lite_server/module.py index 739739b74c238ebe531cbfc3cf984da2af2024ea..5e7be439d8c6a22a43fa020982e7bc1709639499 100644 --- a/hub_module/modules/image/face_detection/pyramidbox_lite_server/module.py +++ b/hub_module/modules/image/face_detection/pyramidbox_lite_server/module.py @@ -28,6 +28,7 @@ class PyramidBoxLiteServer(hub.Module): self.default_pretrained_model_path = os.path.join( self.directory, "pyramidbox_lite_server_face_detection") self._set_config() + self.processor = self def _set_config(self): """ diff --git a/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/module.py b/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/module.py index 59270c64ebb003c6a4f277b1c86abef713e90168..06cc0f3cee0101b6806f6dfa4545cbe3a8babddc 100644 --- a/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/module.py +++ b/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/module.py @@ -37,6 +37,7 @@ class PyramidBoxLiteServerMask(hub.Module): else: self.face_detector = face_detector_module self._set_config() + self.processor = self def _set_config(self): """ diff --git a/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/processor.py b/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/processor.py index 3e3bf1515143cfbcd1dccda71b03cd3b0cf1e77b..61c7be6addd3c1791d1c8e46d0bcd58dcf93e8c9 100644 --- a/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/processor.py +++ b/hub_module/modules/image/face_detection/pyramidbox_lite_server_mask/processor.py @@ -94,12 +94,12 @@ def draw_bounding_box_on_image(save_im_path, output_data): box_fill = (255) text_fill = (0) - draw.rectangle( - xy=(bbox['left'], bbox['top'] - (textsize_height + 5), - bbox['left'] + textsize_width + 10, bbox['top'] - 3), - fill=box_fill) - draw.text( - xy=(bbox['left'], bbox['top'] - 15), text=text, fill=text_fill) + draw.rectangle( + xy=(bbox['left'], bbox['top'] - (textsize_height + 5), + bbox['left'] + textsize_width + 10, bbox['top'] - 3), + fill=box_fill) + draw.text( + xy=(bbox['left'], bbox['top'] - 15), text=text, fill=text_fill) image.save(save_im_path) diff --git a/hub_module/modules/image/object_detection/ssd_mobilenet_v1_pascal/module.py b/hub_module/modules/image/object_detection/ssd_mobilenet_v1_pascal/module.py index a433d7790ff30c50fd2b7819949f0f889c7162ac..8ee1e7edf1ced1b0e1d616e780d91a7f4e53cb20 100644 --- a/hub_module/modules/image/object_detection/ssd_mobilenet_v1_pascal/module.py +++ b/hub_module/modules/image/object_detection/ssd_mobilenet_v1_pascal/module.py @@ -21,7 +21,7 @@ from ssd_mobilenet_v1_pascal.data_feed import reader @moduleinfo( name="ssd_mobilenet_v1_pascal", - version="1.1.0", + version="1.1.1", type="cv/object_detection", summary="SSD with backbone MobileNet_V1, trained with dataset Pasecal VOC.", author="paddlepaddle", @@ -275,7 +275,7 @@ class SSDMobileNetv1(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/ssd_vgg16_300_coco2017/module.py b/hub_module/modules/image/object_detection/ssd_vgg16_300_coco2017/module.py index 6027bcd0d22659a47597ef2fe6615e79751adce2..7310bf04d9e208e36e4d72a25ccc93b8a001012f 100644 --- a/hub_module/modules/image/object_detection/ssd_vgg16_300_coco2017/module.py +++ b/hub_module/modules/image/object_detection/ssd_vgg16_300_coco2017/module.py @@ -21,7 +21,7 @@ from ssd_vgg16_300_coco2017.data_feed import reader @moduleinfo( name="ssd_vgg16_300_coco2017", - version="1.0.0", + version="1.0.1", type="cv/object_detection", summary="SSD with backbone VGG16, trained with dataset COCO.", author="paddlepaddle", @@ -264,7 +264,7 @@ class SSDVGG16(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/ssd_vgg16_512_coco2017/module.py b/hub_module/modules/image/object_detection/ssd_vgg16_512_coco2017/module.py index bdf252ad1001faddf51d92d7c57305b4d6441b8a..e7246367c5e9e895e8d840f37bb66ce8e0ad2b17 100644 --- a/hub_module/modules/image/object_detection/ssd_vgg16_512_coco2017/module.py +++ b/hub_module/modules/image/object_detection/ssd_vgg16_512_coco2017/module.py @@ -21,7 +21,7 @@ from ssd_vgg16_512_coco2017.data_feed import reader @moduleinfo( name="ssd_vgg16_512_coco2017", - version="1.0.0", + version="1.0.1", type="cv/object_detection", summary="SSD with backbone VGG16, trained with dataset COCO.", author="paddlepaddle", @@ -278,7 +278,7 @@ class SSDVGG16_512(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/module.py b/hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/module.py index f3ebbda18e5c3aee46027564c902d8e825b38481..1b693fb27c695e14030e81c3f3e623f7d04a0651 100644 --- a/hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/module.py +++ b/hub_module/modules/image/object_detection/yolov3_darknet53_coco2017/module.py @@ -21,7 +21,7 @@ from yolov3_darknet53_coco2017.yolo_head import MultiClassNMS, YOLOv3Head @moduleinfo( name="yolov3_darknet53_coco2017", - version="1.1.0", + version="1.1.1", type="CV/object_detection", summary= "Baidu's YOLOv3 model for object detection, with backbone DarkNet53, trained with dataset coco2017.", @@ -270,7 +270,7 @@ class YOLOv3DarkNet53Coco2017(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/module.py b/hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/module.py index 1258b7572ec743cb0a4355e34ca7953c6eb8f435..7630d372801bb633d765ba966bb2eaa458df58e0 100644 --- a/hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/module.py +++ b/hub_module/modules/image/object_detection/yolov3_darknet53_pedestrian/module.py @@ -21,7 +21,7 @@ from yolov3_darknet53_pedestrian.yolo_head import MultiClassNMS, YOLOv3Head @moduleinfo( name="yolov3_darknet53_pedestrian", - version="1.0.0", + version="1.0.1", type="CV/object_detection", summary= "Baidu's YOLOv3 model for pedestrian detection, with backbone DarkNet53.", @@ -280,7 +280,7 @@ class YOLOv3DarkNet53Pedestrian(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/module.py b/hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/module.py index e1a064cdf3bf7ea4e339212823cd0ab23c698d23..801228f6b576b34bfcf8de04d0f56f956d17c709 100644 --- a/hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/module.py +++ b/hub_module/modules/image/object_detection/yolov3_darknet53_vehicles/module.py @@ -21,7 +21,7 @@ from yolov3_darknet53_vehicles.yolo_head import MultiClassNMS, YOLOv3Head @moduleinfo( name="yolov3_darknet53_vehicles", - version="1.0.0", + version="1.0.1", type="CV/object_detection", summary= "Baidu's YOLOv3 model for vehicles detection, with backbone DarkNet53.", @@ -280,7 +280,7 @@ class YOLOv3DarkNet53Vehicles(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/module.py b/hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/module.py index 674e6aa5d6d24eeecc43f10b949faef3067e99a1..659cde37ca481f524ebd33daedc80eb3adc24b3a 100644 --- a/hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/module.py +++ b/hub_module/modules/image/object_detection/yolov3_mobilenet_v1_coco2017/module.py @@ -21,7 +21,7 @@ from yolov3_mobilenet_v1_coco2017.yolo_head import MultiClassNMS, YOLOv3Head @moduleinfo( name="yolov3_mobilenet_v1_coco2017", - version="1.0.0", + version="1.0.1", type="CV/object_detection", summary= "Baidu's YOLOv3 model for object detection with backbone MobileNet_V1, trained with dataset COCO2017.", @@ -270,7 +270,7 @@ class YOLOv3MobileNetV1Coco2017(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py b/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py index 54f4f427956e77c551abc12247b55685e1c1cc59..14e31fdd667d3fc9eaa78f185d0c57984fb7ebe0 100644 --- a/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py +++ b/hub_module/modules/image/object_detection/yolov3_resnet34_coco2017/module.py @@ -21,7 +21,7 @@ from yolov3_resnet34_coco2017.yolo_head import MultiClassNMS, YOLOv3Head @moduleinfo( name="yolov3_resnet34_coco2017", - version="1.0.0", + version="1.0.1", type="CV/object_detection", summary= "Baidu's YOLOv3 model for object detection with backbone ResNet34, trained with dataset coco2017.", @@ -272,7 +272,7 @@ class YOLOv3ResNet34Coco2017(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu, diff --git a/hub_module/modules/image/object_detection/yolov3_resnet50_vd_coco2017/module.py b/hub_module/modules/image/object_detection/yolov3_resnet50_vd_coco2017/module.py index 2e65abfe863e60d65d7f57664c3f2c201ab2a968..5c25f17cd37df1a7e67dba76ade582ddbc0b89e5 100644 --- a/hub_module/modules/image/object_detection/yolov3_resnet50_vd_coco2017/module.py +++ b/hub_module/modules/image/object_detection/yolov3_resnet50_vd_coco2017/module.py @@ -21,7 +21,7 @@ from yolov3_resnet50_vd_coco2017.yolo_head import MultiClassNMS, YOLOv3Head @moduleinfo( name="yolov3_resnet50_vd_coco2017", - version="1.0.0", + version="1.0.1", type="CV/object_detection", summary= "Baidu's YOLOv3 model for object detection with backbone ResNet50, trained with dataset coco2017.", @@ -274,7 +274,7 @@ class YOLOv3ResNet50Coco2017(hub.Module): self.add_module_config_arg() self.add_module_input_arg() args = self.parser.parse_args(argvs) - results = self.face_detection( + results = self.object_detection( paths=[args.input_path], batch_size=args.batch_size, use_gpu=args.use_gpu,