add demo

applications/tools/first-order-demo.py

+#  Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
+#
+#Licensed under the Apache License, Version 2.0 (the "License");
+#you may not use this file except in compliance with the License.
+#You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#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 argparse
+
+import paddle
+from ppgan.apps.first_order_predictor import FirstOrderPredictor
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--config", default=None, help="path to config")
+parser.add_argument("--weight_path",
+                    default=None,
+                    help="path to checkpoint to restore")
+parser.add_argument("--source_image", type=str, help="path to source image")
+parser.add_argument("--driving_video", type=str, help="path to driving video")
+parser.add_argument("--output", default='output', help="path to output")
+parser.add_argument("--relative",
+                    dest="relative",
+                    action="store_true",
+                    help="use relative or absolute keypoint coordinates")
+parser.add_argument(
+    "--adapt_scale",
+    dest="adapt_scale",
+    action="store_true",
+    help="adapt movement scale based on convex hull of keypoints")
+
+parser.add_argument(
+    "--find_best_frame",
+    dest="find_best_frame",
+    action="store_true",
+    help=
+    "Generate from the frame that is the most alligned with source. (Only for faces, requires face_aligment lib)"
+)
+
+parser.add_argument("--best_frame",
+                    dest="best_frame",
+                    type=int,
+                    default=None,
+                    help="Set frame to start from.")
+parser.add_argument("--cpu", dest="cpu", action="store_true", help="cpu mode.")
+
+parser.set_defaults(relative=False)
+parser.set_defaults(adapt_scale=False)
+
+if __name__ == "__main__":
+    args = parser.parse_args()
+
+    if args.cpu:
+        paddle.set_device('cpu')
+
+    predictor = FirstOrderPredictor(output=args.output,
+                                    weight_path=args.weight_path,
+                                    config=args.config,
+                                    relative=args.relative,
+                                    adapt_scale=args.adapt_scale,
+                                    find_best_frame=args.find_best_frame,
+                                    best_frame=args.best_frame)
+    predictor.run(args.source_image, args.driving_video)