diff --git a/pyqt/generate_target/gui_target.py b/pyqt/generate_target/gui_target.py index ea260217913291ff1329ec1cf1a3354ed38b1141..2d43d4a435a4f7521c2489ab0ecb8c2a47e27112 100644 --- a/pyqt/generate_target/gui_target.py +++ b/pyqt/generate_target/gui_target.py @@ -108,9 +108,13 @@ class TargetGenerationWindow(QtWidgets.QMainWindow, Ui_MainWindow): vol_initial = np.zeros((dim_z, dim_y, dim_x)) if strategy == 'Shapes': - target = tbuild.generate_with_shapes(objl, vol_initial, param_list) + mask_list = [] + for fname in param_list: # load masks + mask = cm.read_array(fname) + mask_list.append(mask) + target = tbuild.generate_with_shapes(objl, vol_initial, mask_list) else: - param_list = list(map(int, param_list)) # convert the radius list from str to int + param_list = list(map(int, param_list)) # convert the radius list from str to int target = tbuild.generate_with_spheres(objl, vol_initial, param_list) tbuild.display('Saving target ...')