Callbacks

1
2
3
4
# na: number of anchors 每一个predict head上的anchor数 = 3
a = (len(anchors[0]) // 2) if isinstance(anchors, list) else anchors
# no: number of outputs 每一个predict head层的输出channel = anchors * (classes + 5) = 75(VOC)
no = na * (nc + 5)
1
2
model = Model(cfg or ckpt['model'].yaml, ch=3, nc=nc, anchors=hyp.get('anchors')).to(device)  # create
中hyp.get('anchors')???

8.12 # Trainloader

yolov5_mask项目调试