faster rcnn 如何修改测试配置文件

sinat_34139917 2016-05-10 08:49:18
已经用faster rcnn训练好了模型,但是测试的时候一直出现如下错误,请问这是什么原因造成的? test.prototxt除了修改cls_score和bbox_pred之外,还需要修改什么地方? 谢谢了
...全文
3449 6 打赏 收藏 转发到动态 举报
写回复
用AI写文章
6 条回复
切换为时间正序
请发表友善的回复…
发表回复
weixin_43821303 2019-05-06
  • 打赏
  • 举报
回复
请问迭代次数要在哪里修改
blackmoo 2017-01-13
  • 打赏
  • 举报
回复
我告诉你为什么吧,你的end2end都改成2类输出了,但是运行demo会默认选择alt_opt中的test文件运行,所以你要改掉对应的demo中的代码,选择用faster_end2end 的test结构文件
qq_24037421 2016-09-07
  • 打赏
  • 举报
回复
应该还有num_class,至于cls_score不记得要改啊,一般修改的时候,按ctrl+f寻找关键数字,如类的种类21那么要修改21的自己84的,就是类的数量乘以4
sinat_28498195 2016-06-16
  • 打赏
  • 举报
回复
看样子你似乎是用的某个数据集自带的配置文件,像这样:

你要确定你的test.prototxt与你当初训练使用的train.prototxt相应位置的参数是否一致,仔细检查一下是不是test.protxt文件用错了!这个文件是不需要自己手动修改的。
Wenxy1 2016-05-22
  • 打赏
  • 举报
回复
没用过此库,帮你顶。
【资源说明】 1、该资源包括项目的全部源码,下载可以直接使用! 2、本项目适合作为计算机、数学、电子信息等专业的课程设计、期末大作业和毕设项目,作为参考资料学习借鉴。 3、本资源作为“参考资料”如果需要实现其他功能,需要能看懂代码,并且热爱钻研,自行调试。 基于Faster_RCNN网络模型的车辆、行人及交通信号目标检测算法python源码+数据集+项目报告+详细注释.zip ## 环境配置: * Python3.6/3.7/3.8 * Pytorch1.6(注意:必须是1.6.0或以上,因为使用官方提供的混合精度训练1.6.0后才支持) * pycocotools(Linux:```pip install pycocotools```; Windows:```pip install pycocotools-windows```(不需要额外安装vs)) * Ubuntu或Centos(不建议Windows) * 最好使用GPU训练 * 详细环境配置见```requirements.txt``` ## 文件结构: ``` ├── backbone: 特征提取网络,可以根据自己的要求选择 ├── network_files: Faster R-CNN网络(包括Fast R-CNN以及RPN等模块) ├── train_utils: 训练验证相关模块(包括cocotools) ├── my_dataset.py: 自定义dataset用于读取VOC数据集 ├── train_mobilenet.py: 以MobileNetV2做为backbone进行训练 ├── train_resnet50_fpn.py: 以resnet50+FPN做为backbone进行训练 ├── train_multi_GPU.py: 针对使用多GPU的用户使用 ├── predict.py: 简易的预测脚本,使用训练好的权重进行预测测试 ├── validation.py: 利用训练好的权重验证/测试数据的COCO指标,并生成record_mAP.txt文件 └── pascal_voc_classes.json: pascal_voc标签文件 ``` ## 预训练权重下载地址(下载后放入backbone文件夹中): * MobileNetV2 backbone: https://download.pytorch.org/models/mobilenet_v2-b0353104.pth * ResNet50+FPN backbone: https://download.pytorch.org/models/fasterrcnn_resnet50_fpn_coco-258fb6c6.pth * 注意,下载的预训练权重记得要重命名,比如在train_resnet50_fpn.py中读取的是```fasterrcnn_resnet50_fpn_coco.pth```文件, 不是```fasterrcnn_resnet50_fpn_coco-258fb6c6.pth``` ## 数据集,本例程使用的是PASCAL VOC2012数据集 * Pascal VOC2012 train/val数据集下载地址:http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar * 使用ResNet50+FPN以及迁移学习在VOC2012数据集上得到的权重: 链接:https://pan.baidu.com/s/1ifilndFRtAV5RDZINSHj5w 提取码:dsz8 ## 训练方法 * 确保提前准备好数据集 ......

427

社区成员

发帖
与我相关
我的任务
社区描述
非技术问题的乐园
社区管理员
  • 非技术区社区
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧