Anchor Number about yolo-fastestv2

[Yuanye-F]

Solve the problem.

In utils/utils.py:

line 300 :
from
return torch.stack((wv, hv), 2).repeat(1,1,3).reshape(h, w, cfg["anchor_num"], -1).to(device)
to
return torch.stack((wv, hv), 2).repeat(1,1,cfg["anchor_num"]).reshape(h, w, cfg["anchor_num"], -1).to(device)

line 326:
from
c = c.repeat(1, 1, 3, 1)
to
c = c.repeat(1, 1, cfg["anchor_num"], 1)