TensorRT-CenterNet icon indicating copy to clipboard operation
TensorRT-CenterNet copied to clipboard

请教

Open mcmingchang opened this issue 6 years ago • 10 comments

里面那两个模型怎么训练的呀

mcmingchang avatar Nov 08 '19 09:11 mcmingchang

里面那两个模型怎么训练的呀

你好,里面的模型训练,你可以参考centerface,不过他还没开源,估计后期会有人开源。我自己调的在通用VOC上map不是很好,就没有上传。

CaoWGG avatar Nov 09 '19 02:11 CaoWGG

@CaoWGG 你好,不知道你发布的模型是否可以检测车辆,后续我是否可以根据我的数据进行迁移学习,谢谢。

EvaneSnow avatar Nov 11 '19 06:11 EvaneSnow

@EvaneSnow 你好,你可以通过 CenterNet 来训练(选择没有deform conv的backbone(例如dlav0))。训练好后将模型转到onnx模型即可。转换方法参考https://github.com/CaoWGG/TensorRT-CenterNet/issues/1

CaoWGG avatar Nov 11 '19 07:11 CaoWGG

@EvaneSnow CenterNet

CaoWGG avatar Nov 11 '19 07:11 CaoWGG

@CaoWGG 感谢啊,我有不懂地再问你。

EvaneSnow avatar Nov 11 '19 07:11 EvaneSnow

@EvaneSnow 不客气

CaoWGG avatar Nov 11 '19 07:11 CaoWGG

@CaoWGG 可不可以加个联系方式,以后方便交流

EvaneSnow avatar Nov 11 '19 07:11 EvaneSnow

@EvaneSnow 微信: 15706290529

CaoWGG avatar Nov 11 '19 07:11 CaoWGG

我用模型可视化工具完整还原了centerface的模型,在原版的centernet上训练,为什么hm和reg的loss都能下降,wh的loss就一直没有下降呢

mcmingchang avatar Nov 11 '19 07:11 mcmingchang

@mcmingchang 你好,我没有遇到你这个情况,没有调试信息,我也不知道什么问题额。

CaoWGG avatar Nov 11 '19 07:11 CaoWGG