search

bubbliiiing / efficientdet-pytorch

这是一个efficientdet-pytorch的源码，可以用于训练自己的模型。
MIT License
309 stars 61 forks source link

get_dr_txt.py出现问题 #14

Open liuxi1234 opened 3 years ago

liuxi1234 commented 3 years ago

RuntimeError: Error(s) in loading state_dict for EfficientDetBackbone: size mismatch for classifier.header.pointwise_conv.conv.weight: copying a param with shape torch.Size([396, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([810, 64, 1, 1]). size mismatch for classifier.header.pointwise_conv.conv.bias: copying a param with shape torch.Size([396]) from checkpoint, the shape in current model is torch.Size([810]). 为什么会出现上面的维度不匹配的问题

bubbliiiing commented 3 years ago

问题汇总

liuxi1234 commented 3 years ago

问题汇总里面没找到啊

bubbliiiing commented 3 years ago

model_path和classes_path要修改

wangshushu0213 commented 3 years ago

博主给的数据集是不是都是coco数据集，标注格式是voc这个意思？所以classes_path应该是coco那个？

bubbliiiing commented 3 years ago

我给的权重是coco的，数据集格式是voc的