Inquiry About main_pretrain.py and Details on Intensity Decoder and Edge Decoder in the Paper

Davidczy / Uni4Eye_pp

11 stars 1 forks source link

Inquiry About main_pretrain.py and Details on Intensity Decoder and Edge Decoder in the Paper #2

Closed yanglei1111 closed 5 months ago

yanglei1111 commented 8 months ago

Could you please provide more details about the main_pretrain.py code? I couldn't find that file in the project.Especially regarding the intensity decoder and edge decoder (Lssl = λ1Li + λ2Le) mentioned in the paper.

Davidczy commented 8 months ago

Sorry, my code is currently undergoing journal expansion. It will take some time before the code is fully released. However， my code is based on MAE, if you find some codes missing in my project, you can refer to the implementation. For the dual decoder, I just copy the structure of Decoder in MAE. The only difference is the reconstruction target for these two branches.