Remove check

This commit is contained in:
Florent Guiotte 2020-05-31 11:35:55 +02:00
parent 732060d160
commit 769c07373c

View File

@ -42,11 +42,6 @@ def run(gt, rasters, coords, remove, attributes, adjacency='4', filtering='direc
X_raw = np.moveaxis(np.array(list(eap.vectorize())), 0, -1) X_raw = np.moveaxis(np.array(list(eap.vectorize())), 0, -1)
y_raw = gti y_raw = gti
if np.isnan(X_raw).any():
raise Exception('Found {} Nan on {}'.format(np.isnan(X_raw).sum(),
coordsi))
# Remove unwanted label X, y # Remove unwanted label X, y
lbl = np.ones_like(y_raw, dtype=np.bool) lbl = np.ones_like(y_raw, dtype=np.bool)
for l in remove if remove else []: for l in remove if remove else []: