import torch
from models.experimental import attempt_load
model = attempt_load('xxxx.pth')
m = model.module.model[-1] if hasattr(model, 'module') else model.model[-1]
#print(m.anchor_grid)
for i in m.anchor_grid:
#print(i)
print(i[0][0][0][0][:])
print(i[0][1][0][0][:])
print(i[0][2][0][0][:])
anchor获取 pytorch
于 2024-05-30 11:34:59 首次发布