| 157 | |
| 158 | |
| 159 | class InferenceOptions(BaseOptions): |
| 160 | def __init__(self): |
| 161 | super().__init__() |
| 162 | |
| 163 | def initialize(self, parser): |
| 164 | super().initialize(parser) |
| 165 | parser.add_argument("--ref_path", |
| 166 | default=None, type=str,help='ref') |
| 167 | parser.add_argument('--aud_path', |
| 168 | default=None, type=str, help='audio') |
| 169 | parser.add_argument('--emo', |
| 170 | default=None, type=str, help='emotion', choices=['angry', 'disgust', 'fear', 'happy', 'neutral', 'sad', 'surprise']) |
| 171 | parser.add_argument('--no_crop', |
| 172 | action = 'store_true', help = 'not using crop') |
| 173 | parser.add_argument('--res_video_path', |
| 174 | default=None, type=str, help='res video path') |
| 175 | parser.add_argument('--ckpt_path', |
| 176 | default="/home/nvadmin/workspace/taek/float-pytorch/checkpoints/float.pth", type=str, help='checkpoint path') |
| 177 | parser.add_argument('--res_dir', |
| 178 | default="./results", type=str, help='result dir') |
| 179 | return parser |
| 180 | |
| 181 | |
| 182 | if __name__ == '__main__': |