重新训练faster rcnn调用的预训练模型

2017-12-12 09:00:15 2.2k 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

【mmdetection】使用cascade-rcnn、faster-rcnn训练自定义的coco数据集开始这篇文章主要是记录我第一次使用mmdetection工具包里面的cascade-rcnn训练自己模型的经过训练环境本次是使用的系统是linux的ubuntu20.04 cuda=10.2 cudnn=7.6.5 anaconda=3（用于创建虚拟环境，后续的训练、测试模型也在这个虚拟环境中） (以上软件安装可以参考其他博客) pytroch=1.5.0 torchvision=0.6.0 cudatoolkit=10.2.89 mmcv-full=1.3.1 mmdetection=v2.11 阅读详情

最近小鱼碰到一个难题，就是我要重新训练一个ZF.model来给训练faster rcnn的时候进行调用。调用预训练模型是因为这个已经训练好的模型是参数最优了，但如果我想重新自己训练一个模型，比如就是ZF.model，也可以是VGG16.model，要怎么重新训练出预训练的模型？求各位大牛为小鱼解答一下，谢谢。

...全文