未验证 提交 3dcabc48 编写于 作者: W wangna11BD 提交者: GitHub

add model export doc for msvsr (#608)

上级 060c1839
...@@ -119,6 +119,18 @@ ...@@ -119,6 +119,18 @@
To train or test other VSR model, you can find the config file of the corresponding VSR model in the ``PaddleGAN/configs``, then change the config file in the command to the config file of corresponding VSR model. To train or test other VSR model, you can find the config file of the corresponding VSR model in the ``PaddleGAN/configs``, then change the config file in the command to the config file of corresponding VSR model.
### 1.2.3 Model export
Take the msvsr model as an example, ``inputs_size`` is the input size, ``model_name`` is the name of the exported model, and ``model_path`` is the path of the model weight.
```
python tools/export_model.py -c configs/msvsr_reds.yaml --inputs_size="1,2,3,180,320" --model_name inference --load model_path
```
### 1.2.4 Model inference
Take the msvsr model as an example.
```
python tools/inference.py --model_type msvsr -c configs/msvsr_reds.yaml --output_path output_dir
```
## 1.3 Results ## 1.3 Results
The experimental results are evaluated on RGB channel. The experimental results are evaluated on RGB channel.
......
...@@ -118,6 +118,19 @@ ...@@ -118,6 +118,19 @@
训练或测试其他视频超分模型,可以在``PaddleGAN/configs``文件夹下找到对应模型的配置文件,将命令中的配置文件改成该视频超分模型的配置文件即可。 训练或测试其他视频超分模型,可以在``PaddleGAN/configs``文件夹下找到对应模型的配置文件,将命令中的配置文件改成该视频超分模型的配置文件即可。
### 1.2.3 模型导出
以msvsr模型为例,``inputs_size``为模型输入size,``model_name``为导出模型的命名,``model_path``为模型权重的路径.
```
python tools/export_model.py -c configs/msvsr_reds.yaml --inputs_size="1,2,3,180,320" --model_name inference --load model_path
```
### 1.2.4 模型推理
以msvsr模型为例
```
python tools/inference.py --model_type msvsr -c configs/msvsr_reds.yaml --output_path output_dir
```
## 1.3 实验结果展示 ## 1.3 实验结果展示
实验数值结果是在 RGB 通道上进行评估。 实验数值结果是在 RGB 通道上进行评估。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册