raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

zorder of bboxes changed by adding them all together in a patch collection

Browse Source
This commit is contained in:
Till Raab 2023-10-20 13:22:55 +02:00
parent 1bb1de442f
commit ebf9690ad3
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
data/generate_dataset.py
View File

@ -19,7 +19,7 @@ import os
from IPython import embed from IPython import embed
from matplotlib.patches import Rectangle from matplotlib.patches import Rectangle
# from matplotlib.collections import PatchCollection from matplotlib.collections import PatchCollection
def load_data(folder): def load_data(folder):
@ -155,22 +155,36 @@ def main(args):
right_time_bound[enu] = np.nan right_time_bound[enu] = np.nan
else: else:
right_time_bound[enu] = rise_end_t[0] right_time_bound[enu] = rise_end_t[0]
left_time_bound -= 2
right_time_bound *= 2
lower_freq_bound -= 2
upper_freq_bound += 0.2*rise_size_oi
print(f'f0: {lower_freq_bound}') print(f'f0: {lower_freq_bound}')
print(f'f1: {upper_freq_bound}') print(f'f1: {upper_freq_bound}')
print(f't0: {left_time_bound}') print(f't0: {left_time_bound}')
print(f't1: {right_time_bound}') print(f't1: {right_time_bound}')
# Create patch collection with specified colour/alpha # Create patch collection with specified colour/alpha
bbox_col = []
for enu in range(len(left_time_bound)): for enu in range(len(left_time_bound)):
if np.isnan(right_time_bound[enu]): if np.isnan(right_time_bound[enu]):
continue continue
ax.add_patch( bbox_col.append(
Rectangle((left_time_bound[enu], lower_freq_bound[enu]), Rectangle((left_time_bound[enu], lower_freq_bound[enu]),
(right_time_bound[enu] - left_time_bound[enu]), (right_time_bound[enu] - left_time_bound[enu]),
(upper_freq_bound[enu] - lower_freq_bound[enu]), (upper_freq_bound[enu] - lower_freq_bound[enu]),
fill=False, color="white", linewidth=2) fill=False, color="white", linewidth=2)
) )
# ax.add_patch(
# Rectangle((left_time_bound[enu], lower_freq_bound[enu]),
# (right_time_bound[enu] - left_time_bound[enu]),
# (upper_freq_bound[enu] - lower_freq_bound[enu]),
# fill=False, color="white", linewidth=2)
# )
coll = PatchCollection(bbox_col, zorder=10)
ax.add_collection(coll)
plt.show() plt.show()