#906·CenterNet

运行 test.py 文件时出现问题,当我尝试在自己的数据集中使用 CenterNet 时。麻烦各位帮忙解答一下,我更换了 backbone,在训练好模型后,进行测试时说我的预训练参数没有加载。

作者: xls888创建于 2021年4月15日更新于 2025年8月26日

Drop parameter conv5.0.weight.If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset. Drop parameter conv5.1.weight.If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset. Drop parameter conv5.1.bias.If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset. Drop parameter conv5.1.running_mean.If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset. Drop parameter conv5.1.running_var.If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset. Drop parameter conv5.1.num_batches_tracked.If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset. No param conv1_2.0.weight.If you see this, your model does not fully load the pre-trained weight. Please make sure you have correctly specified --arch xxx or set the correct --num_classes for your own dataset.

内容来源: xingyizhou/CenterNet