Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
498de35
1
Parent(s):
e2ec19d
Removed save images
Browse files
app.py
CHANGED
|
@@ -502,9 +502,9 @@ def blend_details(input_image, relit_image, masked_image):
|
|
| 502 |
final_image = torch.clamp(final_image, 0, 1)
|
| 503 |
|
| 504 |
# Save intermediate results for debugging
|
| 505 |
-
tensor2pil(segmentation_mask).save("output/segmentation_mask.png")
|
| 506 |
-
tensor2pil(high_freq_details).save("output/high_freq_details.png")
|
| 507 |
-
tensor2pil(masked_details).save("output/masked_details.png")
|
| 508 |
|
| 509 |
# Save final result
|
| 510 |
final_image_pil = tensor2pil(final_image)
|
|
|
|
| 502 |
final_image = torch.clamp(final_image, 0, 1)
|
| 503 |
|
| 504 |
# Save intermediate results for debugging
|
| 505 |
+
# tensor2pil(segmentation_mask).save("output/segmentation_mask.png")
|
| 506 |
+
# tensor2pil(high_freq_details).save("output/high_freq_details.png")
|
| 507 |
+
# tensor2pil(masked_details).save("output/masked_details.png")
|
| 508 |
|
| 509 |
# Save final result
|
| 510 |
final_image_pil = tensor2pil(final_image)
|