想把其他backbone放进faster里面试一试,但是没有找到合适的预训练模型,用coco数据集该怎么从头训练。
1条回答 默认 最新
- 爱晚乏客游 2021-03-18 16:42关注
加载原来的预训练模型,一般每一层网络都会有一个id,找到你需要的id(这里可以使用https://netron.app/来查看你的预训练模型的结构),这些就不怎么需要变换,然后把不需要的去掉,换上你的网络层,连接好整个网络,设置下你的网络层参数就可以开始训练了。可以参考下这个文章:https://blog.csdn.net/litt1e/article/details/103584016
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报