更换骨干网络
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)如果不想用注释掉就好
感谢您的回复，问题已解决！😁