From 02f83aa3530c3052e98a0180d35c3d75daf6c7d7 Mon Sep 17 00:00:00 2001
From: emoebel <emmanuel.moebel@inria.fr>
Date: Wed, 13 Apr 2022 15:28:42 +0200
Subject: [PATCH] gui_target: corrected bug "AttributeError: 'str' object has
no attribute 'shape'" that happened with shape strategy
---
pyqt/generate_target/gui_target.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/pyqt/generate_target/gui_target.py b/pyqt/generate_target/gui_target.py
index ea26021..2d43d4a 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 ...')
--
GitLab