faster-rcnn-pytorch icon indicating copy to clipboard operation
faster-rcnn-pytorch copied to clipboard

没有找到frcnn_weights.pth文件,未能使用预训练权重进行预测

Open zhengze98 opened this issue 3 years ago • 2 comments

大大您好,没有找到您在【a.使用预训练权重】这一节中提到的frcnn_weights.pth文件,我只能将网盘中的voc_weights_resnet.pth放进了model_data下,然后运行predict.py,输入img/street.jpg 进行测试。结果出现了维度不匹配的问题(确定cls_calsses等路径是对的,因为都没改过) 是因为voc_weights_resnet.pth是骨干权重,跟预测模型使用的权重不是一码事吗,也就是说不能混用? 还是说是其他的原因导致的... 冒昧打扰您,非常抱歉

zhengze98 avatar Apr 21 '22 04:04 zhengze98

你frcnn.py文件里面的model_data和classes路径改了没?我记得我也遇到过这样的问题

swlwhut avatar Apr 25 '22 08:04 swlwhut

frcnn_weights.pth是以前的名字……之前更新了vgg主干后就弃用了。 你如果使用这个权重是针对voc数据集的,所以是voc_classes.txt

bubbliiiing avatar Apr 26 '22 00:04 bubbliiiing