加载预训练的yolov4小砝码时出错

Question

我正在尝试使用YOLOv4-tiny创建对象检测器，并且从下面的链接中获得了权重和cfg文件。

• weights
• cfg

当我加载权重时，出现以下运行时错误：

RuntimeError：形状'[256、384、3、3]'对于大小为756735的输入无效

我的代码：

config_path='yolov4-tiny.cfg'
weights_path='yolov4-tiny.weights'
class_path='coco.names'
img_size=416
conf_thres=0.8 #confidence threshold
nms_thres=0.4 #non max suppression thresh

model=Darknet(config_path,img_size=img_size)
model.load_weights(weights_path)
model.cuda()
model.eval()
classes=utils.load_classes(class_path)