simonfranken/mmp_wise2526_franksim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

renaming

Browse Source
This commit is contained in:
franksim
2025-11-11 11:08:07 +01:00
parent 7245042b54
commit 56e21a1e54
25 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mmp/a4/dataset.py
View File

@@ -120,7 +120,7 @@ def get_dataloader(
return dataloader
def calculate_max_coverage(loader, min_iou):
def calculate_coverage(loader, min_iou):
"""
@param loader: DataLoader object.
@param min_iou: Minimum IoU overlap to count a ground truth box as covered.
@@ -254,7 +254,7 @@ def draw_positive_boxes(
def main():
anchor_grid = get_anchor_grid(
anchor_widths=[8, 16, 35, 32, 64, 96, 128, 144, 150, 160, 192, 224],
anchor_widths=[8, 16, 32, 64, 96, 128, 160, 192],
aspect_ratios=[1 / 3, 1 / 2, 3 / 5, 2 / 3, 3 / 4, 1, 4 / 3, 5 / 3, 2, 2.5, 3],
num_rows=28,
num_cols=28,
@@ -270,7 +270,7 @@ def main():
anchor_grid=anchor_grid,
)
# print(calculate_max_coverage(dataloader, 0.7))
# print(calculate_coverage(dataloader, 0.7))
for img, label, img_id in islice(dataloader, 12):
draw_positive_boxes(