Faster R-CNN如何与ResNet50搭配检测?

最近想要把Faster R-CNN的特征提取部分的VGG16网络改成ResNet50,尝试简单的将models/pascal_voc/faster_rcnn_end2end里面的模型文件改了一下,采取end2end的训练方式。改完之后发现不断报错,要么内存错(8G显存GPU),要么别的错,想知道有没有大神改过啊?求教一下~
已邀请:

数学爱好者26 - it男

赞同来自:

你是直接把vgg16给替换掉?还是进行裁剪?

要回复问题请先登录注册