Just update in the plotting method

lrm22005 · Nov 27, 2023 · ad734c6 · ad734c6
1 parent b3b4b4d
commit ad734c6
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.