From 7366a63b1e0d216f29f9d11794231d31093823a2 Mon Sep 17 00:00:00 2001 From: chenjian Date: Tue, 8 Mar 2022 17:39:39 +0800 Subject: [PATCH] fix according to review --- modules/image/image_processing/prnet/api.py | 10 ++++++---- .../image/image_processing/prnet/module.py | 20 ++++++++++--------- modules/image/image_processing/prnet/util.py | 1 + 3 files changed, 18 insertions(+), 13 deletions(-) diff --git a/modules/image/image_processing/prnet/api.py b/modules/image/image_processing/prnet/api.py index 17cb4e79..60f91cf4 100644 --- a/modules/image/image_processing/prnet/api.py +++ b/modules/image/image_processing/prnet/api.py @@ -1,10 +1,12 @@ -import numpy as np import os - -from skimage.io import imread, imsave -from skimage.transform import estimate_transform, warp from time import time + +import numpy as np import paddle +from skimage.io import imread +from skimage.io import imsave +from skimage.transform import estimate_transform +from skimage.transform import warp from .predictor import PosPrediction diff --git a/modules/image/image_processing/prnet/module.py b/modules/image/image_processing/prnet/module.py index 596cece2..a35c81e9 100644 --- a/modules/image/image_processing/prnet/module.py +++ b/modules/image/image_processing/prnet/module.py @@ -11,23 +11,25 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. - -import os import argparse import copy +import os -import paddle -import paddlehub as hub -from paddlehub.module.module import moduleinfo, runnable, serving -import numpy as np import cv2 +import numpy as np +import paddle from skimage.io import imread -from skimage.transform import rescale, resize +from skimage.transform import rescale +from skimage.transform import resize -from .util import base64_to_cv2 +import paddlehub as hub +from .api import PRN from .predictor import PosPrediction +from .util import base64_to_cv2 from .utils.render import render_texture -from .api import PRN +from paddlehub.module.module import moduleinfo +from paddlehub.module.module import runnable +from paddlehub.module.module import serving @moduleinfo(name="prnet", type="CV/", author="paddlepaddle", author_email="", summary="", version="1.0.0") diff --git a/modules/image/image_processing/prnet/util.py b/modules/image/image_processing/prnet/util.py index 9629e72e..92719947 100644 --- a/modules/image/image_processing/prnet/util.py +++ b/modules/image/image_processing/prnet/util.py @@ -1,4 +1,5 @@ import base64 + import cv2 import numpy as np -- GitLab