LiheYoung / ST-PlusPlus

[CVPR 2022] ST++: Make Self-training Work Better for Semi-supervised Semantic Segmentation
https://arxiv.org/abs/2106.05095
MIT License
233 stars 33 forks source link

Some question about the ST code settings #7

Closed chaochao42 closed 3 years ago

chaochao42 commented 3 years ago

In your code, when running "Total stage 1: Supervised training on labeled images (SupOnly)", it gets best model by evaluating on validate dataset. I want to know is it fair to use the information provided by validate dataset to select best model? I think in training stage all you have is labeled and unlabeled training dataset.

LiheYoung commented 3 years ago

Hi, thanks for your question.

Strictly speaking, it is better not to touch the validation set during training, which is actually the testing set in semi-supervised semantic segmentation. However, previous works such as Context-aware Consistency also validate the model every few epochs.

Besides, I want to note that in most settings, the performance of the final epoch is comparable to the best model during training. You can simply use the final model for evaluation.