From 9e22482f74e44ca68558a035008b36db7f55a44f Mon Sep 17 00:00:00 2001 From: wangna11BD <79366697+wangna11BD@users.noreply.github.com> Date: Wed, 9 Nov 2022 18:42:24 +0800 Subject: [PATCH] support cpu infer for vsr (#725) --- applications/tools/video-enhance.py | 6 ++++++ docs/en_US/tutorials/video_restore.md | 1 + docs/zh_CN/industrial_solution/video_restore_cn.md | 1 + 3 files changed, 8 insertions(+) diff --git a/applications/tools/video-enhance.py b/applications/tools/video-enhance.py index e7476a7..6d1b13d 100644 --- a/applications/tools/video-enhance.py +++ b/applications/tools/video-enhance.py @@ -109,9 +109,15 @@ parser.add_argument('--process_order', default='none', nargs='+', help='Process order') +parser.add_argument("--cpu", + dest="cpu", + action="store_true", + help="cpu mode.") if __name__ == "__main__": args = parser.parse_args() + if args.cpu: + paddle.set_device('cpu') orders = args.process_order temp_video_path = None diff --git a/docs/en_US/tutorials/video_restore.md b/docs/en_US/tutorials/video_restore.md index 8365c8f..5022bc0 100644 --- a/docs/en_US/tutorials/video_restore.md +++ b/docs/en_US/tutorials/video_restore.md @@ -12,6 +12,7 @@ python tools/video-enhance.py --input you_video_path.mp4 --process_order DAIN De - `--input (str)`: path of the input video. - `--output (str)`: path of the output video. - `--process_order`: name and order of called model. For example, if the input is `DAIN DeOldify EDVR`, then `DAINPredictor` `DeOldifyPredictor` `EDVRPredictor` will be called in sequence. +- `--cpu`: use cpu inference, GPU inference is used by default. #### Results ![](../../imgs/color_sr_peking.gif) diff --git a/docs/zh_CN/industrial_solution/video_restore_cn.md b/docs/zh_CN/industrial_solution/video_restore_cn.md index 2727a92..df5eebf 100644 --- a/docs/zh_CN/industrial_solution/video_restore_cn.md +++ b/docs/zh_CN/industrial_solution/video_restore_cn.md @@ -16,6 +16,7 @@ python tools/video-enhance.py --input you_video_path.mp4 --process_order DAIN De - `--input (str)`: 输入的视频路径。 - `--output (str)`: 输出的视频路径。 - `--process_order`: 调用的模型名字和顺序，比如输入为 `DAIN DeOldify PPMSVSR`，则会顺序调用 `DAINPredictor` `DeOldifyPredictor` `PPMSVSRPredictor` 。 +- `--cpu`: 开启cpu推理，默认使用GPU推理。
-- GitLab