jhj0517 commited on
Commit
ae0eee9
1 Parent(s): e5d2146

Update docs

Browse files
Files changed (1) hide show
  1. modules/utils.py +2 -2
modules/utils.py CHANGED
@@ -31,8 +31,8 @@ def get_image_files(image_dir: str):
31
  def save_image(image: Union[np.ndarray, str],
32
  output_path: Optional[str] = None,
33
  output_dir: Optional[str] = None):
34
- """Save the image to the output path or output directory. If output directory is provided,
35
- the image will be saved as a numbered image file in the directory."""
36
 
37
  if output_dir is None and output_path is None:
38
  raise ValueError("Either output_path or output_dir should be provided")
 
31
  def save_image(image: Union[np.ndarray, str],
32
  output_path: Optional[str] = None,
33
  output_dir: Optional[str] = None):
34
+ """Save the image to the output path or output directory. If output_dir is provided,
35
+ the image will be saved as a numbered image file name in the directory."""
36
 
37
  if output_dir is None and output_path is None:
38
  raise ValueError("Either output_path or output_dir should be provided")