Method parse_args

config/config.py:205–254

(self)

        self.parser = argparse.ArgumentParser()

    def parse_args(self):
        self.parser.add_argument(
            "--source_channel", type=int, default=3, help="channels of source image"
        )
        self.parser.add_argument(
            "--ref_channel", type=int, default=15, help="channels of reference image"
        )
        self.parser.add_argument(
            "--audio_channel", type=int, default=29, help="channels of audio feature"
        )
        self.parser.add_argument(
            "--mouth_region_size", type=int, default=256, help="help to resize window"
        )
        self.parser.add_argument(
            "--source_video_path",
            default="./asserts/examples/test4.mp4",
            type=str,
            help="path of source video",
        )
        self.parser.add_argument(
            "--source_openface_landmark_path",
            default="./asserts/examples/test4.csv",
            type=str,
            help="path of detected openface landmark",
        )
        self.parser.add_argument(
            "--driving_audio_path",
            default="./asserts/examples/driving_audio_1.wav",
            type=str,
            help="path of driving audio",
        )
        self.parser.add_argument(
            "--pretrained_clip_DINet_path",
            default="./asserts/clip_training_DINet_256mouth.pth",
            type=str,
            help="pretrained model of DINet(clip trained)",
        )
        self.parser.add_argument(
            "--deepspeech_model_path",
            default="./asserts/output_graph.pb",
            type=str,
            help="path of deepspeech model",
        )
        self.parser.add_argument(
            "--res_video_dir",
            default="./asserts/inference_result",
            type=str,
            help="path of generated videos",
        )
        return self.parser.parse_args()