FIX: astype on scalar value not allowed

daf3840a · Gaetano Raffaele · 92b37087 · daf3840a
Commit daf3840a authored 10 months ago by Gaetano Raffaele
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 2 deletions
+2 -2
--- a/OBIA/OBIABase.py
+++ b/OBIA/OBIABase.py
@@ -377,14 +377,14 @@ class OBIABase:
            self.populate_map(tn,obj_id,classes,output_file,compress)

    def true_pred_bypixel(self, labels, predicted_classes, class_field='class'):
-        pred_c = np.zeros(np.max(self.ref_db['orig_label']).astype(int)+1)
+        pred_c = np.zeros(int(np.max(self.ref_db['orig_label']))+1)
        pred_c[labels] = predicted_classes
        support = []
        for tn, t in self.tiled_objects(on_ref=True):
            support.append(t[np.isin(t, labels)])
        support = np.concatenate(support)
        pred = pred_c[support]
-        true_c = np.zeros(np.max(self.ref_db['orig_label']).astype(int)+1)
+        true_c = np.zeros(int(np.max(self.ref_db['orig_label']))+1)
        # ATTENTION: works if "labels" is sorted (as provided by get_reference_...)
        true_c[labels] = self.ref_db.loc[self.ref_db['orig_label'].isin(labels),class_field].to_numpy(dtype=int)
        true = true_c[support]