提交 c48d8210 编写于 作者: C chenjian

modify model name

上级 6af4bec3
# painttransformer # paint_transformer
|模型名称|painttransformer| |模型名称|paint_transformer|
| :--- | :---: | | :--- | :---: |
|类别|图像 - 风格转换| |类别|图像 - 风格转换|
|网络|Paint Transformer| |网络|Paint Transformer|
...@@ -41,7 +41,7 @@ ...@@ -41,7 +41,7 @@
- ### 2、安装 - ### 2、安装
- ```shell - ```shell
$ hub install painttransformer $ hub install paint_transformer
``` ```
- 如您安装时遇到问题,可参考:[零基础windows安装](../../../../docs/docs_ch/get_start/windows_quickstart.md) - 如您安装时遇到问题,可参考:[零基础windows安装](../../../../docs/docs_ch/get_start/windows_quickstart.md)
| [零基础Linux安装](../../../../docs/docs_ch/get_start/linux_quickstart.md) | [零基础MacOS安装](../../../../docs/docs_ch/get_start/mac_quickstart.md) | [零基础Linux安装](../../../../docs/docs_ch/get_start/linux_quickstart.md) | [零基础MacOS安装](../../../../docs/docs_ch/get_start/mac_quickstart.md)
...@@ -52,7 +52,7 @@ ...@@ -52,7 +52,7 @@
- ```shell - ```shell
# Read from a file # Read from a file
$ hub run painttransformer --input_path "/PATH/TO/IMAGE" $ hub run paint_transformer --input_path "/PATH/TO/IMAGE"
``` ```
- 通过命令行方式实现风格转换模型的调用,更多请见 [PaddleHub命令行指令](../../../../docs/docs_ch/tutorial/cmd_usage.rst) - 通过命令行方式实现风格转换模型的调用,更多请见 [PaddleHub命令行指令](../../../../docs/docs_ch/tutorial/cmd_usage.rst)
...@@ -61,7 +61,7 @@ ...@@ -61,7 +61,7 @@
- ```python - ```python
import paddlehub as hub import paddlehub as hub
module = hub.Module(name="painttransformer") module = hub.Module(name="paint_transformer")
input_path = ["/PATH/TO/IMAGE"] input_path = ["/PATH/TO/IMAGE"]
# Read from a file # Read from a file
module.style_transfer(paths=input_path, output_dir='./transfer_result/', use_gpu=True) module.style_transfer(paths=input_path, output_dir='./transfer_result/', use_gpu=True)
...@@ -92,7 +92,7 @@ ...@@ -92,7 +92,7 @@
- 运行启动命令: - 运行启动命令:
- ```shell - ```shell
$ hub serving start -m painttransformer $ hub serving start -m paint_transformer
``` ```
- 这样就完成了一个油画风格转换的在线服务API的部署,默认端口号为8866。 - 这样就完成了一个油画风格转换的在线服务API的部署,默认端口号为8866。
...@@ -117,7 +117,7 @@ ...@@ -117,7 +117,7 @@
# 发送HTTP请求 # 发送HTTP请求
data = {'images':[cv2_to_base64(cv2.imread("/PATH/TO/IMAGE"))]} data = {'images':[cv2_to_base64(cv2.imread("/PATH/TO/IMAGE"))]}
headers = {"Content-type": "application/json"} headers = {"Content-type": "application/json"}
url = "http://127.0.0.1:8866/predict/painttransformer" url = "http://127.0.0.1:8866/predict/paint_transformer"
r = requests.post(url=url, headers=headers, data=json.dumps(data)) r = requests.post(url=url, headers=headers, data=json.dumps(data))
# 打印预测结果 # 打印预测结果
...@@ -130,5 +130,5 @@ ...@@ -130,5 +130,5 @@
初始发布 初始发布
- ```shell - ```shell
$ hub install painttransformer==1.0.0 $ hub install paint_transformer==1.0.0
``` ```
...@@ -31,13 +31,13 @@ from .util import base64_to_cv2 ...@@ -31,13 +31,13 @@ from .util import base64_to_cv2
@moduleinfo( @moduleinfo(
name="painttransformer", name="paint_transformer",
type="CV/style_transfer", type="CV/style_transfer",
author="paddlepaddle", author="paddlepaddle",
author_email="", author_email="",
summary="", summary="",
version="1.0.0") version="1.0.0")
class painttransformer: class paint_transformer:
def __init__(self): def __init__(self):
self.pretrained_model = os.path.join(self.directory, "paint_best.pdparams") self.pretrained_model = os.path.join(self.directory, "paint_best.pdparams")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册