set seeds

utils/segmentation.py

@@ -4,6 +4,7 @@ import torch
 import scipy.sparse as sp
 import sys
 import os
+import random
 from zipfile import ZipFile
 from .plotting import plot_side_by_side_comparison
 
@@ -12,6 +13,13 @@ from models.HybridGNet2IGSC import Hybrid
 
 hybrid = None
 
+def seed_everything(seed=42):
+    random.seed(seed)
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    if torch.cuda.is_available():
+        torch.cuda.manual_seed_all(seed)
+
 def scipy_to_torch_sparse(scp_matrix):
     values = scp_matrix.data
     indices = np.vstack((scp_matrix.row, scp_matrix.col))
@@ -215,7 +223,9 @@ def segment(input_img, noise_std=0.0):
         img_corr = np.clip(img_corr + noise, 0.0, 1.0)
 
     # Predict landmarks
+    seed_everything(123)
     means_orig, stds_orig = predict_landmarks(img_orig)
+    seed_everything(123)
     means_corr, stds_corr = predict_landmarks(img_corr)
 
     # Save landmarks and masks