Update src/config/inference_config.py

src/config/inference_config.py

@@ -35,7 +35,7 @@ class InferenceConfig(PrintableConfig):
 
     input_shape: Tuple[int, int] = (256, 256)  # input shape
     output_format: Literal['mp4', 'gif'] = 'mp4'  # output video format
-    output_fps: int = 30  # fps for output video
+    output_fps: int = 20  # fps for output video
     crf: int = 15  # crf for output video
 
     flag_write_result: bool = True  # whether to write output video
@@ -43,7 +43,7 @@ class InferenceConfig(PrintableConfig):
     mask_crop: ndarray = cv2.imread(make_abs_path('../utils/resources/mask_template.png'), cv2.IMREAD_COLOR)
     flag_write_gif: bool = False
     size_gif: int = 256
-    ref_max_shape: int = 1280
+    ref_max_shape: int = 256
     ref_shape_n: int = 2
 
     device_id: int = 0