search

ykdai / Flare7K

Official Implementation of "Flare7K: A Phenomenological Nighttime Flare Removal Dataset"
https://ykdai.github.io/projects/Flare7K
Other
119 stars 16 forks source link

Can you please provide lens flare segmentation network training code with its checkpoint #22

Closed GopiRajuMatta closed 2 months ago

GopiRajuMatta commented 3 months ago

Hello!

In Flare7k++ paper, you mentioned lens flare segmentation as one of the potential applications. Can you please provide the network(PSPNet) training code with its checkpoint, I am planning to use it for one of my works.

Otherwise, can you please provide gt segmentation masks, so that I can train the network by myself.

Thank you Gopi

ykdai commented 2 months ago

We are using mmseg for training. However, we do not have plans to provide its checkpoint. For gt segmentation masks, you can set a threshold for the glare and streak annotations for training.

GopiRajuMatta commented 2 months ago

Sure, will try it out, thank you for the response..