AlibabaResearch / efficientteacher

A Supervised and Semi-Supervised Object Detection Library for YOLO Series
GNU General Public License v3.0
820 stars 147 forks source link

伪标签问题和自己预训练模型问题 #87

Open wangsun1996 opened 1 year ago

wangsun1996 commented 1 year ago

前辈好,我有两个问题想请教: 1.在设置debug=True后,存下来的图片一直都是空图(图片上并没有叠加伪标签),请问这个正常吗(COCO训练过程打印的mAP还挺高)?还是哪里没设置对?在设置debug为true后,我根据报错提示(提示存储路径不存在)修改了图片保存路径 2.我先试了COCO数据集,能正常运行。然后在自己的7类数据上进行SSOD训练报错,后来发现好像在加载预训练模型时,需要在将pt转为ET时确保pt模型的类别数与后续自己数据集的类别数相同,否则会在后续训练加载模型参数时提示维度不匹配。这里等于必须用对应的pt模型吗?

BowieHsu commented 1 year ago

@wangsun1996 您好,针对1,如果保存的是空图,那说明应该是没有伪标签生成的。针对2,我当时应该解除过这个bug了,我们这边直接用coco和obj365预训练的pt进行ssod也是可以加载的

Meoooww commented 1 year ago

那如果是伪标签过多,应该调整哪一个参数呢

shy-7 commented 1 year ago

@wangsun1996 您好,针对1,如果保存的是空图,那说明应该是没有伪标签生成的。针对2,我当时应该解除过这个bug了,我们这边直接用coco和obj365预训练的pt进行ssod也是可以加载的

请问针对2,解除了bug的代码在哪里呢,没有找到

fengfasheng commented 11 months ago

前辈好,我有两个问题想请教: 1.在设置debug=True后,存下来的图片一直都是空图(图片上并没有叠加伪标签),请问这个正常吗(COCO训练过程打印的mAP还挺高)?还是哪里没设置对?在设置debug为true后,我根据报错提示(提示存储路径不存在)修改了图片保存路径 2.我先试了COCO数据集,能正常运行。然后在自己的7类数据上进行SSOD训练报错,后来发现好像在加载预训练模型时,需要在将pt转为ET时确保pt模型的类别数与后续自己数据集的类别数相同,否则会在后续训练加载模型参数时提示维度不匹配。这里等于必须用对应的pt模型吗?

你好,我想问一下这边保存路径应该修改成什么呢