Update run_diffueraser.py

#5
Files changed (1) hide show
  1. run_diffueraser.py +1 -1
run_diffueraser.py CHANGED
@@ -12,7 +12,7 @@ def main():
12
  parser.add_argument('--input_video', type=str, default="examples/example3/video.mp4", help='Path to the input video')
13
  parser.add_argument('--input_mask', type=str, default="examples/example3/mask.mp4" , help='Path to the input mask')
14
  parser.add_argument('--video_length', type=int, default=10, help='The maximum length of output video')
15
- parser.add_argument('--mask_dilation_iter', type=int, default=8, help='Adjust it to change the degree of mask expansion')
16
  parser.add_argument('--max_img_size', type=int, default=960, help='The maximum length of output width and height')
17
  parser.add_argument('--save_path', type=str, default="results" , help='Path to the output')
18
  parser.add_argument('--ref_stride', type=int, default=10, help='Propainter params')
 
12
  parser.add_argument('--input_video', type=str, default="examples/example3/video.mp4", help='Path to the input video')
13
  parser.add_argument('--input_mask', type=str, default="examples/example3/mask.mp4" , help='Path to the input mask')
14
  parser.add_argument('--video_length', type=int, default=10, help='The maximum length of output video')
15
+ parser.add_argument('--mask_dilation_iter', type=int, default=2, help='Adjust it to change the degree of mask expansion')
16
  parser.add_argument('--max_img_size', type=int, default=960, help='The maximum length of output width and height')
17
  parser.add_argument('--save_path', type=str, default="results" , help='Path to the output')
18
  parser.add_argument('--ref_stride', type=int, default=10, help='Propainter params')