LiheYoung / UniMatch

[CVPR 2023] Revisiting Weak-to-Strong Consistency in Semi-Supervised Semantic Segmentation
https://arxiv.org/abs/2208.09910
MIT License
478 stars 60 forks source link

The quality of the pseudo-labels in the first few epochs #65

Closed xiaona123456 closed 1 year ago

xiaona123456 commented 1 year ago

When training a new field of the dataset, such as medical, Can you tell me if the pseudo-labels generated in the first few epochs will be of poorer quality, and will they be sufficient for training? I haven't run experiments with my data yet, can you give me some theoretical guidance? Thanks a lot!

with torch.no_grad():
        model.eval()

        pred_u_w_mix = model(img_u_w_mix).detach()
        conf_u_w_mix = pred_u_w_mix.softmax(dim=1).max(dim=1)[0]
        mask_u_w_mix = pred_u_w_mix.argmax(dim=1)
LiheYoung commented 1 year ago

It is normal that pseudo labels in the early stages are poor. They will gradually improve as the training goes. We also use a confidence threshold to discard uncertain pseudo labels, which are mostly poor.

LiheYoung commented 1 year ago

Closing due to inactivity.