Hotfix quality range from 0-100

s0md3v · Jul 25, 2023 · 1422849 · 1422849
1 parent 96c8f1c
commit 1422849
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -52,9 +52,9 @@ options:
   --reference-frame-number REFERENCE_FRAME_NUMBER                            number of the reference frame
   --similar-face-distance SIMILAR_FACE_DISTANCE                              face distance used for recognition
   --temp-frame-format {jpg,png}                                              image format used for frame extraction
-  --temp-frame-quality [1-100]                                               image quality used for frame extraction
+  --temp-frame-quality [0-100]                                               image quality used for frame extraction
   --output-video-encoder {libx264,libx265,libvpx-vp9,h264_nvenc,hevc_nvenc}  encoder used for the output video
-  --output-video-quality [1-100]                                             quality used for the output video
+  --output-video-quality [0-100]                                             quality used for the output video
   --max-memory MAX_MEMORY                                                    maximum amount of RAM in GB
   --execution-provider {cpu} [{cpu} ...]                                     available execution provider (choices: cpu, ...)
   --execution-threads EXECUTION_THREADS                                      number of execution threads

diff --git a/roop/core.py b/roop/core.py
@@ -45,9 +45,9 @@ def parse_args() -> None:
     program.add_argument('--reference-frame-number', help='number of the reference frame', dest='reference_frame_number', type=int, default=0)
     program.add_argument('--similar-face-distance', help='face distance used for recognition', dest='similar_face_distance', type=float, default=0.85)
     program.add_argument('--temp-frame-format', help='image format used for frame extraction', dest='temp_frame_format', default='png', choices=['jpg', 'png'])
-    program.add_argument('--temp-frame-quality', help='image quality used for frame extraction', dest='temp_frame_quality', type=int, default=0, choices=range(100), metavar='[0-100]')
+    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]')
     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'])
-    program.add_argument('--output-video-quality', help='quality used for the output video', dest='output_video_quality', type=int, default=35, choices=range(100), metavar='[0-100]')
+    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]')
     program.add_argument('--max-memory', help='maximum amount of RAM in GB', dest='max_memory', type=int)
     program.add_argument('--execution-provider', help='available execution provider (choices: cpu, ...)', dest='execution_provider', default=['cpu'], choices=suggest_execution_providers(), nargs='+')
     program.add_argument('--execution-threads', help='number of execution threads', dest='execution_threads', type=int, default=suggest_execution_threads())