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

applies linter changes

Browse Source
This commit is contained in:
franksim
2025-11-07 11:21:28 +01:00
parent b159d76517
commit 5c7e1364cb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mmp/a4/label_grid.py
View File

@@ -26,9 +26,9 @@ def get_label_grid(
label_grid = np.empty(anchor_grid.shape[:-1], dtype=bool) label_grid = np.empty(anchor_grid.shape[:-1], dtype=bool)
for (width, ratio, row, col), item in np.ndenumerate(anchor_grid): for (width, ratio, row, col), item in np.ndenumerate(anchor_grid):
for gt in gts: for gt in gts:
iou = iou(item, gt) calculated_iou = iou(item, gt)
label_grid[width, ratio, row, col] = False label_grid[width, ratio, row, col] = False
if iou >= min_iou: if calculated_iou >= min_iou:
label_grid[width, ratio, row, col] = True label_grid[width, ratio, row, col] = True
break break
return label_grid return label_grid