GlassyWu / AECR-Net

Contrastive Learning for Compact Single Image Dehazing, CVPR2021
339 stars 48 forks source link

How to select multiple negative samples in the paper #4

Open my721ncepu opened 3 years ago

my721ncepu commented 3 years ago

Great work! I have some question about your paper: 1.How to make a positive sample with multiple negative samples to compute contastive loss? 2.How to select multiple negative samples in paper?

Looking forward to your reply

GlassyWu commented 3 years ago

For Q1, each negative samples will be compute with the positive sample, and we adopt the average distance. For Q2, we select negative samples from same batch images, and we only use 10 negative samples, because of the limited GPU memory size.