Merge pull request #9 from lrm22005/Luis

Just update in the plotting method
lrm22005 · Nov 27, 2023 · 74bc620 · 74bc620
2 parents 4b44afa + ad734c6
commit 74bc620
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/project_NVIL.py b/project_NVIL.py
@@ -216,21 +216,6 @@ def imshow(image, ax=None, title=None, normalize=True):
 
 
 
-
-fig, axes = plt.subplots(figsize=(20,15), ncols=6)
-
-for i, batch in enumerate(train_loader):
-    images, label, names = batch["data"], batch["label"], batch["segment_name"]
-    ax = axes[i]
-#     helper.imshow(images[ii], ax=ax, normalize=False)
-    imshow(images[i], ax=ax, normalize=False)
-    if i==6:
-        break
-
-
-
-
-
 class PriorNet(nn.Module):
     """
     An NN that parameterises a prior distribution.