diff --git a/python_package/src/onlinemid/roi/roi.py b/python_package/src/onlinemid/roi/roi.py index 7ff8b4762c82401c69b7ff58fa4e431599223e78..aafa594000b5e33069253123ca7b1ab8fc5a1e4a 100644 --- a/python_package/src/onlinemid/roi/roi.py +++ b/python_package/src/onlinemid/roi/roi.py @@ -29,11 +29,12 @@ def make_onlineroi_from_roi_array(roi_array): while roi_ravel[index+length] is True: length +=1 - compressed_data.add(np.array([index, length])) + compressed_data.add(np.array([index, length], dtype=np.uint64)) print(compressed_data.get()) index += length else: index += 1 + print(compressed_data.get()) return compressed_data.get() online_roi_data = dict() @@ -43,13 +44,13 @@ def make_onlineroi_from_roi_array(roi_array): # reshape this online_roi_data into a np array. online_roi_data_max_length = 0 - for key, online_roi_data in online_roi_data.values(): + for key, online_roi_data in online_roi_data.items(): this_size = online_roi_data.size if this_size > online_roi_data_max_length: online_roi_data_max_length = this_size online_roi = np.zeros(shape=(num_rois, online_roi_data_max_length), dtype=np.uint64) - for key, online_roi_data in online_roi_data.values(): + for key, online_roi_data in online_roi_data.items(): this_size = online_roi_data.size online_roi[int(key),:this_size] = online_roi_data