--- a +++ b/utils/preprocessing/reduce_single_phase.py @@ -0,0 +1,17 @@ +import numpy as np +import os + + +in_dir = "pp_train_128_ven" + +for i in os.listdir(in_dir): + if "img.npy" in i: + print (i) + arr = np.load(os.path.join(in_dir,i)) + + print ("shape now ",arr.shape) + + arr = arr[:,:,:,0] + + arr = np.save(os.path.join(in_dir,i),arr) +print ("done")