更换骨干网络
Opened this issue · 2 comments
您好,我想请教一个问题,我已经将骨干网络替换成CSPDarket-53,但是不想下载初始权重文件,希望网络从0开始训练(对比试验的特殊需求),我应该修改backbone.py和train.py中的哪一部分呢,删除哪些类呢?望您在百忙之中给予指点,谢谢!
您好,我想请教一个问题,我已经将骨干网络替换成CSPDarket-53,但是不想下载初始权重文件,希望网络从0开始训练(对比试验的特殊需求),我应该修改backbone.py和train.py中的哪一部分呢,删除哪些类呢?望您在百忙之中给予指点,谢谢!
train.py不用改。我使用的resnet预训练模型是pytorch官方的,其下载与导入在backbone.py的301和303行,
state_dict = load_checkpoint(self.backbone_name)
# 导入权重
clasify_model = load_state_dict(clasify_model, state_dict)
如果不想用注释掉就好
您好,我想请教一个问题,我已经将骨干网络替换成CSPDarket-53,但是不想下载初始权重文件,希望网络从0开始训练(对比试验的特殊需求),我应该修改backbone.py和train.py中的哪一部分呢,删除哪些类呢?望您在百忙之中给予指点,谢谢!
train.py不用改。我使用的resnet预训练模型是pytorch官方的,其下载与导入在backbone.py的301和303行,
state_dict = load_checkpoint(self.backbone_name)
# 导入权重
clasify_model = load_state_dict(clasify_model, state_dict)如果不想用注释掉就好
感谢您的回复,问题已解决!😁