Closed Factoryfromhere closed 2 years ago
我也发现了这个问题,他的网络虽然没用label,但是计算损失时都会用到label
您好,请您仔细阅读代码,计算损失函数包括diceh和ce,here. 哪里用到了没有labeled数据的label了?label_batch[:args.labeled_bs] 指的是仅考虑标注数据标签,无标注数据是伪标签pseudo_sup.
loss没什么问题,数据处理这块,比如brants2019.py里面这个函数:就对label做了操作
train_mean_teachers_3D + BraTs2019,Dataset默认将所有数据的image和label都读入了,实际semi-supervise大多数数据unlabel,后面的transform也是作用于label