From ffcee4e8b11d3b3ff1678107e393ec850383e9df Mon Sep 17 00:00:00 2001 From: "Michael Pilosov, PhD" Date: Fri, 23 Feb 2024 16:45:22 +0000 Subject: [PATCH] image resolution saving default --- callbacks.py | 2 +- check.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/callbacks.py b/callbacks.py index 72f6c6a..39a4309 100644 --- a/callbacks.py +++ b/callbacks.py @@ -25,7 +25,7 @@ class SaveImageCallback(Callback): # else: # version = 0 fname = Path(pl_module.trainer.logger.log_dir) / Path(f"e{epoch:04d}") - create_circle(pl_module, fname=fname) + create_circle(pl_module, fname=fname, dpi=300, figsize=(6, 6)) # Make sure to set it back to train mode pl_module.train() diff --git a/check.py b/check.py index dc77d57..ff8276b 100644 --- a/check.py +++ b/check.py @@ -61,7 +61,7 @@ def create_circle( def plot_preds( - preds, rgb_values, fname: str, roll: bool = False, dpi: int = 150, figsize=(3, 3) + preds, rgb_values, fname: str, roll: bool = False, dpi: int = 300, figsize=(6, 6) ): if isinstance(preds, torch.Tensor): preds = preds.detach().cpu().numpy()