snicolau commited on
Commit
ba05ea5
β€’
1 Parent(s): de3f5e5

Update lama/saicinpainting/training/visualizers/directory.py

Browse files
lama/saicinpainting/training/visualizers/directory.py CHANGED
@@ -3,8 +3,8 @@ import os
3
  import cv2
4
  import numpy as np
5
 
6
- from saicinpainting.training.visualizers.base import BaseVisualizer, visualize_mask_and_images_batch
7
- from saicinpainting.utils import check_and_warn_input_range
8
 
9
 
10
  class DirectoryVisualizer(BaseVisualizer):
 
3
  import cv2
4
  import numpy as np
5
 
6
+ from lama.saicinpainting.training.visualizers.base import BaseVisualizer, visualize_mask_and_images_batch
7
+ from lama.saicinpainting.utils import check_and_warn_input_range
8
 
9
 
10
  class DirectoryVisualizer(BaseVisualizer):