mirror of https://github.com/vladmandic/automatic
13 lines
460 B
Python
13 lines
460 B
Python
# Auto-anchor utils
|
|
|
|
|
|
def check_anchor_order(m):
|
|
# Check anchor order against stride order for YOLOv5 Detect() module m, and correct if necessary
|
|
a = m.anchor_grid.prod(-1).view(-1) # anchor area
|
|
da = a[-1] - a[0] # delta a
|
|
ds = m.stride[-1] - m.stride[0] # delta s
|
|
if da.sign() != ds.sign(): # same order
|
|
print("Reversing anchor order")
|
|
m.anchors[:] = m.anchors.flip(0)
|
|
m.anchor_grid[:] = m.anchor_grid.flip(0)
|