ali-ghamdan commited on
Commit
9f363d7
Β·
1 Parent(s): ac5fd2f

Update deoldify/visualize.py

Browse files
Files changed (1) hide show
  1. deoldify/visualize.py +2 -2
deoldify/visualize.py CHANGED
@@ -24,7 +24,7 @@ def get_watermarked(pil_image: Image) -> Image:
24
  image = np.dstack([image, np.ones((h, w), dtype="uint8") * 255])
25
  pct = 0.05
26
  full_watermark = cv2.imread(
27
- 'C:/Users/galquja/Desktop/Tools/tools/DeOldify/resource_images/watermark.png', cv2.IMREAD_UNCHANGED
28
  )
29
  (fwH, fwW) = full_watermark.shape[:2]
30
  wH = int(pct * h)
@@ -412,7 +412,7 @@ def get_video_colorizer(render_factor: int = 21) -> VideoColorizer:
412
 
413
 
414
  def get_artistic_video_colorizer(
415
- root_folder: Path = Path('C:/Users/galquja/Desktop/Tools/tools/DeOldify'),
416
  weights_name: str = 'ColorizeArtistic_gen',
417
  results_dir='result_images',
418
  render_factor: int = 35
 
24
  image = np.dstack([image, np.ones((h, w), dtype="uint8") * 255])
25
  pct = 0.05
26
  full_watermark = cv2.imread(
27
+ 'resource_images/watermark.png', cv2.IMREAD_UNCHANGED
28
  )
29
  (fwH, fwW) = full_watermark.shape[:2]
30
  wH = int(pct * h)
 
412
 
413
 
414
  def get_artistic_video_colorizer(
415
+ root_folder: Path = Path('./'),
416
  weights_name: str = 'ColorizeArtistic_gen',
417
  results_dir='result_images',
418
  render_factor: int = 35