TypeError: KnowledgeDistillationSingleStageDetector: __init__() got an unexpected keyword argument 'output_feature'

HikariTJU / LD

Localization Distillation for Object Detection (CVPR 2022, TPAMI 2023)

Apache License 2.0

355 stars 51 forks source link

Open zhangnan12d opened 2 years ago

zhangnan12d commented 2 years ago

$G{R_Y4HJ 5%)AXBWHK%DKAB$ 训练的时候遇到了问题，很抱歉打扰您，实在不知道该怎么办

HikariTJU commented 2 years ago

检查一下你的kd_one_stage.py里的 KnowledgeDistillationSingleStageDetector 有没有这一行

zhangnan12d commented 2 years ago

感谢感谢！已经可以训练了！但是不知道为什么训练出来的效果特别差（我用的是自定义的SAR船舶检测数据集），是因为epoch太少的缘故吗？希望您能给我一点建议 UNQ EWDHA 6O~V~NIU$ P5X

Zzh-tju commented 2 years ago

给出训练完整log

zhangnan12d commented 2 years ago

Zzh-tju commented 2 years ago

你的数据集是旋转检测框还是水平框？应该是数据集格式有问题（如果是我我会先检查这个）。其次是注意学习率与GPU数量要匹配一下

HikariTJU commented 2 years ago

你先用gfl训一个试试，因为我看到你教师模型是用的coco上预训练的

zhangnan12d commented 2 years ago

20220703_153520.log 感谢两位老师的建议，对我帮助很大。解决方法：1、改用官方的数据集转换脚本，把voc格式转换为coco格式。2、会报loss=nan的错，将默认的学习率0.02改为0.001就可以了（我只有一个gpu）3、导入合适的教师模型