sapphire497 / query-selected-attention

Official implementation for "QS-Attn: Query-Selected Attention for Contrastive Learning in I2I Translation" (CVPR 2022)
75 stars 12 forks source link

The pretrained model and evaluation DRN #2

Open veroveroxie opened 2 years ago

veroveroxie commented 2 years ago

Hi, thanks for your excellent work.

  1. Could you please release the pre-trained models?

  2. Could you also please release your code of evaluation of cityscapes? The PixAcc of your method is super high, and I would like to compare and cite your work.

  3. About the SWD, may I ask which code are you using? I tried google and I can only get https://github.com/koshian2/swd-pytorch . It uses tensors as input and there can be many ways of generating tensors from images.

Again, thanks for your work! It is very interesting!

sapphire497 commented 2 years ago

Hi!

  1. The pre-trained models will be released in a few days.
  2. I use the official codes in https://github.com/fyu/drn, prepare your synthesis and label images, and run segment.py to calculate the pixAcc and mAP.
  3. You can download the 'evaluation.zip' from https://github.com/microsoft/CoCosNet/issues/25, and run cal_sliced_wasserstein.py to calculate the SWD.
veroveroxie commented 2 years ago
  1. That would be very great! Thanks, looking forward to your models.
  2. Could you please tell me which pretrained model do you use? the pretrained d-22 or d-105-ms? Did you resize the image and labels to 256x128 or 256x256 or 128x256?
  3. Thanks!
sapphire497 commented 2 years ago

I use d-22 model and resize the images to 256x128.

veroveroxie commented 2 years ago

Hi, I used your cat2dog checkpoint to generate images. I can get FID 80, which is the same of the paper. But I used SWD,

I can only get following results

Horse2zebra image

Cat2dog image

Is it a typo in your paper? Or could you please tell me how to compute Swd for cat2dog?

haoren55555 commented 2 years ago

Hi,could you please publish your pretrained segmentation model on cityscapes? thanks!