kai-wen-yang / CD-VAE

[NeurIPS 2021] "Class-Disentanglement and Applications in Adversarial Detection and Defense"
MIT License
43 stars 8 forks source link

Could you please share the pretrained detection model for ImageNet? #10

Open csjunjun opened 1 year ago

kai-wen-yang commented 1 year ago

Now I could not find the imagenet pretrained model, but the implementation for imagenet is similar to cifar. The only difference is we use VQ-VAE for imagenet and I have provided the VQ-VAE code. Would please implement it by yourself and concat me if you have any questions?