search

bubbliiiing / retinanet-pytorch

这是一个retinanet-pytorch的源码，可以用于训练自己的模型。
MIT License
171 stars 27 forks source link

有关改为Resnet101/152的问题 #16

Open speechle55 opened 2 years ago

speechle55 commented 2 years ago
image

我把代码中的phi改成了101代表的数字，并且也在model_data中放入了resnet101的模型，请问这个是什么原因啊，我还需要改哪里吗

speechle55 commented 2 years ago

我看之前的回答说是修改主干，只是多了几个重复的层，请问具体是哪里呢

bubbliiiing commented 2 years ago

你载入了错误的权值……看一下权值载入的部分

speechle55 commented 2 years ago

阿这。。。难道不是把model_path改为下载的resnet101然后把phi改成3就可以吗

bubbliiiing commented 2 years ago

……你看主干的预训练怎么加载的。

arkerman commented 1 year ago

resnet18的预训练权重在哪里下载呀

arkerman commented 1 year ago

@bubbliiiing