请教
里面那两个模型怎么训练的呀
里面那两个模型怎么训练的呀
你好,里面的模型训练,你可以参考centerface,不过他还没开源,估计后期会有人开源。我自己调的在通用VOC上map不是很好,就没有上传。
@CaoWGG 你好,不知道你发布的模型是否可以检测车辆,后续我是否可以根据我的数据进行迁移学习,谢谢。
@EvaneSnow 你好,你可以通过 CenterNet 来训练(选择没有deform conv的backbone(例如dlav0))。训练好后将模型转到onnx模型即可。转换方法参考https://github.com/CaoWGG/TensorRT-CenterNet/issues/1
@EvaneSnow CenterNet
@CaoWGG 感谢啊,我有不懂地再问你。
@EvaneSnow 不客气
@CaoWGG 可不可以加个联系方式,以后方便交流
@EvaneSnow 微信: 15706290529
我用模型可视化工具完整还原了centerface的模型,在原版的centernet上训练,为什么hm和reg的loss都能下降,wh的loss就一直没有下降呢
@mcmingchang 你好,我没有遇到你这个情况,没有调试信息,我也不知道什么问题额。