S0soo / tf-faster-rcnn-ghostnet

This is a demo about Faster R-CNN use GhostNet
4 stars 0 forks source link

您好!非常感谢您所做的工作,非常棒! #1

Open tianyuluan opened 4 years ago

tianyuluan commented 4 years ago

不知是否可以分享一下您的预训练参数 <pretrained_model = r'F:\GhostNet\ghostnet\models\ghostnet_checkpoint'>

S0soo commented 4 years ago

您好,GhostNet的预训练模型来自https://github.com/huawei-noah/ghostnet/tree/master/models

tianyuluan commented 4 years ago

好像这个预训练模型不能用于您更改的ghosnet_faster_rcnn,我看到了您在ghosnet官方repo上提的问题,可以分享一下您的解决办法吗,谢谢🙏

S0soo commented 4 years ago

好像这个预训练模型不能用于您更改的ghosnet_faster_rcnn,我看到了您在ghosnet官方repo上提的问题,可以分享一下您的解决方法吗,谢谢🙏

我想知道预训练模型不能用于ghostnet_faster_rcnn的具体情况是什么,我很抱歉如果不使用预训练文件,超参数的设置可能需要进行大量的搜索,这段时间我在忙于我的毕业论文和一些项目,并没有再继续进行尝试

tianyuluan commented 4 years ago

好像这个预训练模型不能用于您更改的ghosnet_faster_rcnn,我看到了您在ghosnet官方repo上提的问题,可以分享一下您的解决方法吗,谢谢🙏

我想知道预训练模型不能用于ghostnet_faster_rcnn的具体情况是什么,我很抱歉如果不使用预训练文件,超参数的设置可能需要进行大量的搜索,这段时间我在忙于我的毕业论文和一些项目,并没有再继续进行尝试

您是没有进行尝试是吗?我按照您所说的步骤进行了尝试,但是报错 no variables to restore(差不多是这个意思),然后我在deberk(tf_faster_rcnn)中寻求答案,我感觉应该就是华为给出的ghostnet的预训练模型与您更改的模型不匹配造成的,我们是否可以只对ghostnet网络本身进行预先的训练,然后一起训练fasterrcnn-ghostnet? 祝你毕业答辩顺利!

S0soo commented 4 years ago

应该是可以的,我在自己的数据集上就是用了官方提供的预训练文件,但是为了能够应用在faster r-cnn上,我对ghostnet模型进行了细微的改变,主要体现在stride上,所以我只使用了部分的预训练权重,当然全部使用也是可以的,我想知道你的faster r-cnn是根据dbk编译安装的,还是直接使用了这个demo的。