faster-rcnn-pytorch
faster-rcnn-pytorch copied to clipboard
没有找到frcnn_weights.pth文件,未能使用预训练权重进行预测
大大您好,没有找到您在【a.使用预训练权重】这一节中提到的frcnn_weights.pth文件,我只能将网盘中的voc_weights_resnet.pth放进了model_data下,然后运行predict.py,输入img/street.jpg 进行测试。结果出现了维度不匹配的问题(确定cls_calsses等路径是对的,因为都没改过) 是因为voc_weights_resnet.pth是骨干权重,跟预测模型使用的权重不是一码事吗,也就是说不能混用? 还是说是其他的原因导致的... 冒昧打扰您,非常抱歉
你frcnn.py文件里面的model_data和classes路径改了没?我记得我也遇到过这样的问题
frcnn_weights.pth是以前的名字……之前更新了vgg主干后就弃用了。 你如果使用这个权重是针对voc数据集的,所以是voc_classes.txt