lorenmt / reco

The implementation of "Bootstrapping Semantic Segmentation with Regional Contrast" [ICLR 2022].
https://shikun.io/projects/regional-contrast
Other
162 stars 25 forks source link

how to apply reco loss to supervised segmentation #24

Closed hxshust closed 2 years ago

lorenmt commented 2 years ago

I thought train_sup.py should give an example on this?

hxshust commented 2 years ago

Hi,thanks for sharing your impressive code. As I see in your object page, ReCo can easily be added to modern supervised and semi-supervised segmentation methods without changing the training pipeline, with no additional cost at inference time. To incorporate ReCo, we simply add an additional representation head on top of the feature encoder of a segmentation network, and apply the ReCo loss to this representation using the active sampling strategy.

hxshust commented 2 years ago

without additional modifications to the network structure?I thought an extra branch was needed, outside the segmented branch, use the encoder's feature vector with the ground truth to calculate the reco loss

hxshust commented 2 years ago

thanks anyway,I will check the train_sup.py again,: )