作者:aRuis | 来源:互联网 | 2023-08-31 15:32
想把其他backbone放进faster里面试一试,但是没有找到合适的预训练模型,用coco数据集该怎么从头训练。加载原来的预训练模型,一般每一层网络都会有一个id,找到你需要的i
想把其他backbone放进faster里面试一试,但是没有找到合适的预训练模型,用coco数据集该怎么从头训练。
加载原来的预训练模型,一般每一层网络都会有一个id,找到你需要的id(这里可以使用https://netron.app/来查看你的预训练模型的结构),这些就不怎么需要变换,然后把不需要的去掉,换上你的网络层,连接好整个网络,设置下你的网络层参数就可以开始训练了。可以参考下这个文章:https://blog.csdn.net/litt1e/article/details/103584016