diff --git a/modules/image/image_processing/prnet/api.py b/modules/image/image_processing/prnet/api.py index 17cb4e7905719576910c19b33a12059441d42194..60f91cf40b1fb7b4171f1a10b06c02e169292ce3 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 596cece246dfb1dd5716d6fcaccf7c7540354e81..a35c81e9cc2b3376beec3fbb9132e0199d02b8ee 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 9629e72e5efa502eec1377cc1417322312468587..927199476050e27e036078902c21bf92d2b475da 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