tuan2308 commited on
Commit
29b1aaa
1 Parent(s): cfc908e

Update roop/core.py

Browse files
Files changed (1) hide show
  1. roop/core.py +8 -7
roop/core.py CHANGED
@@ -41,14 +41,14 @@ def parse_args() -> None:
41
  program.add_argument('--reference-face-position', help='position of the reference face', dest='reference_face_position', type=int, default=0)
42
  program.add_argument('--reference-frame-number', help='number of the reference frame', dest='reference_frame_number', type=int, default=0)
43
  program.add_argument('--similar-face-distance', help='face distance used for recognition', dest='similar_face_distance', type=float, default=0.85)
44
- program.add_argument('--temp-frame-format', help='image format used for frame extraction', dest='temp_frame_format', default='png', choices=['jpg', 'png'])
45
  program.add_argument('--temp-frame-quality', help='image quality used for frame extraction', dest='temp_frame_quality', type=int, default=0, choices=range(101), metavar='[0-100]')
46
- program.add_argument('--output-video-encoder', help='encoder used for the output video', dest='output_video_encoder', default='libx264', choices=['libx264', 'libx265', 'libvpx-vp9', 'h264_nvenc', 'hevc_nvenc'])
47
- program.add_argument('--output-video-quality', help='quality used for the output video', dest='output_video_quality', type=int, default=35, choices=range(101), metavar='[0-100]')
48
- program.add_argument('--max-memory', help='maximum amount of RAM in GB', dest='max_memory', type=int)
49
- program.add_argument('--execution-provider', help='available execution provider (choices: cpu, ...)', dest='execution_provider', default=['cpu'], choices=suggest_execution_providers(), nargs='+')
50
- program.add_argument('--execution-threads', help='number of execution threads', dest='execution_threads', type=int, default=suggest_execution_threads())
51
- program.add_argument('-v', '--version', action='version', version=f'{roop.metadata.name} {roop.metadata.version}')
52
 
53
  args = program.parse_args()
54
 
@@ -219,3 +219,4 @@ def run() -> None:
219
  else:
220
  window = ui.init(start, destroy)
221
  window.mainloop()
 
 
41
  program.add_argument('--reference-face-position', help='position of the reference face', dest='reference_face_position', type=int, default=0)
42
  program.add_argument('--reference-frame-number', help='number of the reference frame', dest='reference_frame_number', type=int, default=0)
43
  program.add_argument('--similar-face-distance', help='face distance used for recognition', dest='similar_face_distance', type=float, default=0.85)
44
+ program.add.add_argument('--temp-frame-format', help='image format used for frame extraction', dest='temp_frame_format', default='png', choices=['jpg', 'png'])
45
  program.add_argument('--temp-frame-quality', help='image quality used for frame extraction', dest='temp_frame_quality', type=int, default=0, choices=range(101), metavar='[0-100]')
46
+ program.add.add_argument('--output-video-encoder', help='encoder used for the output video', dest='output_video_encoder', default='libx264', choices=['libx264', 'libx265', 'libvpx-vp9', 'h264_nvenc', 'hevc_nvenc'])
47
+ program.add.add_argument('--output-video-quality', help='quality used for the output video', dest='output_video_quality', type=int, default=35, choices=range(101), metavar='[0-100]')
48
+ program.add.add_argument('--max-memory', help='maximum amount of RAM in GB', dest='max_memory', type=int)
49
+ program.add.add_argument('--execution-provider', help='available execution provider (choices: cpu, ...)', dest='execution_provider', default=['cpu'], choices=suggest_execution_providers(), nargs='+')
50
+ program.add.add_argument('--execution-threads', help='number of execution threads', dest='execution_threads', type=int, default=suggest_execution_threads())
51
+ program.add.add_argument('-v', '--version', action='version', version=f'{roop.metadata.name} {roop.metadata.version}')
52
 
53
  args = program.parse_args()
54
 
 
219
  else:
220
  window = ui.init(start, destroy)
221
  window.mainloop()
222
+