Spaces:
Runtime error
Runtime error
Commit
Β·
9f363d7
1
Parent(s):
ac5fd2f
Update deoldify/visualize.py
Browse files- 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 |
-
'
|
| 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('
|
| 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
|