Problem while running the test.py file,when I try to use CenterNet in my own dataset。麻烦各位帮忙解答一下,替换了backbone,我训练好模型之后,进行测试说我预训练参数没有加载.
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. 麻烦各位帮忙解答一下,替换了backbone,我训练好模型之后,进行测试说我预训练参数没有加载,关键这个backbone没有预训练参数啊
Source: xingyizhou/CenterNet