Skip to content

Commit 2bbd99c

Browse files
mariomario
authored andcommitted
reshape I and R before stacking them
1 parent 446f792 commit 2bbd99c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

code/python/cni_toolbox/HGR.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def get_parameters(self, n_batch = 10000,
206206
y = np.sin(np.pi / 4) * R[i]
207207
I[i] = np.mean(gaussian(x, y, S[i], x_coordinates, y_coordinates))
208208

209-
P = np.hstack((I, R))
209+
P = np.hstack((I.reshape(-1,1), R.reshape(-1,1)))
210210
beta = regress(S, P)
211211

212212
for v in np.arange(0, n_mask - n_batch, n_batch):

0 commit comments

Comments
 (0)