【解决bug】RuntimeError: Error(s) in loading state_dict for Net: size mismatch for classifier

飒馍噎

已于 2023-01-18 17:20:25 修改

阅读量3k

点赞数

文章标签： bug python 开发语言

于 2023-01-18 17:18:40 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_44459446/article/details/128728487

版权

在使用DeepSORT处理自定义数据集时，用户遇到了RuntimeError，原因是模型的classifier层权重和偏置的形状不匹配。解决方案是在feature_extractor.py文件中初始化Net类时，指定num_classes为数据集的类别数（NUM_CLASSES）。这允许模型适配用户的数据集类别的数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用deepsort检测自己的数据集时遇到

报错： RuntimeError: Error(s) in loading state_dict for Net:
size mismatch for classifier.4.weight: copying a param with shape torch.Size([23, 256]) from checkpoint, the shape in current model is torch.Size([751, 256]).
size mismatch for classifier.4.bias: copying a param with shape torch.Size([23]) from checkpoint, the shape in current model is torch.Size([751]).

解决方法：

在feature_extractor.py 第12行改成
def __init__(self, model_path, use_cuda=True):
self.net = Net(num_classes=NUM_CLASSES, reid=True)

方法来源：https://github.com/ZQPei/deep_sort_pytorch/issues/134