Closed chenjunson closed 4 years ago
具体代码：elif mdef['type'] == 'yolo': yolo_index += 1 stride = [32, 16, 8, 4] # P5, P4, P3 strides if 'panet' in cfg or 'yolov4' in cfg: # stride order reversed stride = list(reversed(stride)) layers = mdef['from'] if 'from' in mdef else [] modules = YOLOLayer(anchors=mdef['anchors'][mdef['mask']], # anchor list nc=mdef['classes'], # number of classes img_size=img_size, # (416, 416) yolo_index=yolo_index, # 0, 1, 2... layers=layers, # output layers stride=stride[yolo_index]) 只要在stride = [32, 16, 8, 4] 中加上4就行了吧，还需要改别的地方吗？
具体代码：elif mdef['type'] == 'yolo': yolo_index += 1 stride = [32, 16, 8, 4] # P5, P4, P3 strides if 'panet' in cfg or 'yolov4' in cfg: # stride order reversed stride = list(reversed(stride)) layers = mdef['from'] if 'from' in mdef else [] modules = YOLOLayer(anchors=mdef['anchors'][mdef['mask']], # anchor list nc=mdef['classes'], # number of classes img_size=img_size, # (416, 416) yolo_index=yolo_index, # 0, 1, 2... layers=layers, # output layers stride=stride[yolo_index]) 只要在stride = [32, 16, 8, 4] 中加上4就行了吧，还需要改别的地方吗？