search

ColumbiaDVMM / AutoGAN

96 stars 22 forks source link

Pretrained models do not contain checkpoint file #6

Closed carlini closed 4 years ago

carlini commented 4 years ago

The pretrained model obtained from

https://drive.google.com/open?id=1IZZ5QCLbzAmrPDHh7EdOigPCdz46zPc6

does not have any file that matches the path {}{}/checkpoint_{}.pth. Each of the folders has the format

/tmp/autogan_model/monet_autogan                                                                                                                                                                                                                        
/tmp/autogan_model/monet_autogan/latest_net_G.pth                                                                                                                                                                                                       
/tmp/autogan_model/monet_autogan/latest_net_D_A.pth                                                                                                                                                                                                     
/tmp/autogan_model/monet_autogan/test_opt.txt                                                                                                                                                                                                           
/tmp/autogan_model/monet_autogan/train_opt.txt

Are there pretrained checkpoints for running the detector?

spongezhang commented 4 years ago

Hi carlini,

Sorry for the late reply! I was completely overwhelmed by COVID-19 related staffs in the past two months.

Note, the pretrained model in the Google Drive is the pretrained model for AutoGAN, not for the GAN image detector. To train the autogan image detector, you have to use the pretrained AutoGAN model to generate autogan images and use both autogan images and original images to train a classifier using the code in the repo.

Unfortunately, since I've already left Columbia University for about 7 months, the pretrained classifiers are already lost. With autogan images and original images, you should be able to train a classifier within 2 hours.

Certainly let me know if you have any further questions.

Best, Xu

spongezhang commented 4 years ago

No further information. Close the issue.