both padded and non padded imgs

andrevitorelli · Feb 7, 2020 · 35380da · 35380da
1 parent 6ee82bf
commit 35380da
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/blendhunter/run_bh_loop.py b/blendhunter/run_bh_loop.py
@@ -18,7 +18,9 @@
 for j in [5, 14, 18, 26, 35, 40]:
     for i in ['', 1,2,3,4]:
         #Make sure to check the folder hierarchy
-        path = user_home+'/Documents/Cosmostat/Codes/BlendHunter/bh_{}'.format(str(j)+str(i))
+        path = user_home+'/Documents/Cosmostat/Codes/BlendHunter/bh_pad{}'.format(str(j)+str(i))
+        #For non padded images
+        #path = user_home+'/Documents/Cosmostat/Codes/BlendHunter/bh_{}'.format(str(j)+str(i))
         """Make sure each bh folder contains a 'weights' folder"""
         bh = BlendHunter(weights_path=path + '/weights')
 
@@ -41,4 +43,6 @@
         #Save history and results
         #Make sure to check the folder hierarchy
         np.save(path+'/BlendHunterData/test/test/history.npy', hist)
-        np.save(user_home+'/Documents/Cosmostat/Codes/BlendHunter/pad_results/preds_pad{}.npy'.format(str(j)+str(i)), pred_top)
+        np.save(user_home+'/Documents/Cosmostat/Codes/BlendHunter/bh_pad_results/preds_pad{}.npy'.format(str(j)+str(i)), pred_top)
+        #For non padded images
+        #np.save(user_home+'/Documents/Cosmostat/Codes/BlendHunter/bh_results/preds_{}.npy'.format(str(j)+str(i)), pred_top)