diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py index 33c30c1241a06ca7c6da9ba06ad1aa2e98f74c4d..7cd327ae2403509a5486af4eb145a6514e62dcdd 100644 --- a/model_server/base/roiset.py +++ b/model_server/base/roiset.py @@ -290,16 +290,11 @@ class RoiSet(object): # assign labels to object map: for i, roi in enumerate(self): - try: - oc = np.unique( - mask_largest_object( - obmap_patches.iat(i).data - ) - )[-1] - except Exception as e: - print(i) - print(np.unique(mask_largest_object(obmap_patches.iat(i).data))) - raise e + oc = np.unique( + mask_largest_object( + obmap_patches.iat(i).data + ) + )[-1] self._df.loc[roi.Index, 'classify_by_' + name] = oc om[self.acc_obj_ids.data == roi.label] = oc self.object_class_maps[name] = InMemoryDataAccessor(om)