Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The pretrained model and evaluation DRN #2

Open
veroveroxie opened this issue Apr 13, 2022 · 5 comments
Open

The pretrained model and evaluation DRN #2

veroveroxie opened this issue Apr 13, 2022 · 5 comments

Comments

@veroveroxie
Copy link

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
Copy link
Owner

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
Copy link
Author

  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
Copy link
Owner

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

@veroveroxie
Copy link
Author

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
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants