参照这位方法 https://blog.csdn.net/GrayOnDream/article/details/99090247#comments_20256989
问题:出现Traceback (most recent call last):
File "c:/yolo/30/yolov4-tiny-pytorch-master/9.py", line 75, in get_feature()
File "c:/yolo/30/yolov4-tiny-pytorch-master/9.py", line 41, in get_feature net = YoloBody().to(device)
TypeError: init() missing 2 required positional arguments: 'anchors_mask' and 'num_classes'
代码:
class YoloBody(nn.Module):
def init(self, anchors_mask, num_classes=1, phi=0):
super(YoloBody, self).init()
self.phi = phi
self.backbone = darknet53_tiny(None)
self.conv_for_P5 = BasicConv(512,256,1)
self.yolo_headP5 = yolo_head([512, len(anchors_mask[0]) * (5 + num_classes)],256)
self.upsample = Upsample(256,128)
self.yolo_headP4 = yolo_head([256, len(anchors_mask[1]) * (5 + num_classes)],384)
if 1 <= self.phi and self.phi <= 3:
self.feat1_att = attention_blockself.phi - 1
self.feat2_att = attention_blockself.phi - 1
self.upsample_att = attention_blockself.phi - 1