yolodet-pytorch icon indicating copy to clipboard operation
yolodet-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon wuzhihao7788

建议backbone部分都提供加载预训练权值的接口？

Open dsp6414 opened this issue 4 years ago • 1 comments

感觉好像只有resnet可以加载预训练权值

dsp6414 avatar Nov 07 '20 21:11 dsp6414

目前是只有resnet可提供预加载模型。其他模型结构也可通过配置文件from选项变相的预加载模型。但是如果类别不一致的话。可能需要修改一下。最后一层模型输出的的结构。你的建议很好，下一个版本，我可以把此部分修改意见加入进去

wuzhihao7788 avatar Nov 08 '20 01:11 wuzhihao7788