n_rand_views now turned on by default

spoonsso · May 27, 2021 · b5d36e2 · b5d36e2
1 parent b69fda3
commit b5d36e2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/dannce/__init__.py b/dannce/__init__.py
@@ -83,7 +83,7 @@
     "net": None,
     "from_weights": None,
     "dannce_predict_vol_tifdir": None,
-    "n_rand_views": None,
+    "n_rand_views": 0,
     "rand_view_replace": True,
     "multi_gpu_train": False,
 }

diff --git a/dannce/interface.py b/dannce/interface.py
@@ -1003,6 +1003,13 @@ def dannce_train(params: Dict):
         params["rand_view_replace"] = False
         randflag = True
 
+    if params["n_rand_views"] == 0:
+        print("Using default n_rand_views augmentation with {} views and with replacement".format(n_views))
+        print("To disable n_rand_views augmentation, set it to None in the config.")
+        params["n_rand_views"] = n_views
+        params["rand_view_replace"] = True
+
+
     shared_args = {'chan_num': params["chan_num"],
                    'expval': params["expval"],
                    'nvox': params["nvox"]}