search

rightlit / StackGAN-v2-rev

StackGAN-v2 revised and applied demos
MIT License
4 stars 1 forks source link

Can I get the model of flowers? #11

Open kisstherain8677 opened 3 years ago

kisstherain8677 commented 3 years ago

Hi,thanks for your job,can you share the model of flowers? Or can you tell me how to train flowers model? Thank you!

rightlit commented 3 years ago

Hello~ thank you for your interest in my job. please refer to gide in original github(https://github.com/hanzhanggit/StackGAN-v2)

[Data]

[Training]

you can also use config file for flowers model in my github. please check this file : flowers_3stages.yml in https://github.com/rightlit/StackGAN-v2-rev/tree/main/code/cfg/

kisstherain8677 commented 3 years ago

Thanks for your reply! Actually after a few days I've already tried to train flower model by "flowers_3stages.yml".It works but just need some time. Thank you again!

pietarrrrui commented 3 years ago

Thanks for your reply! Actually after a few days I've already tried to train flower model by "flowers_3stages.yml".It works but just need some time. Thank you again!

hi, Do you have pre-trained model? I wanna try for test.

pietarrrrui commented 3 years ago

Hello~ thank you for your interest in my job. please refer to gide in original github(https://github.com/hanzhanggit/StackGAN-v2)

[Data]

  • Download preprocessed char-CNN-RNN text embeddings for birds and flowers and save them to Data/
  • Download the birds and flowers image data. Extract them to Data/birds/ and Data/flowers/, respectively.

[Training]

  • Train a StackGAN-v2 model on the bird (CUB) dataset using our preprocessed embeddings: python main.py --cfg cfg/birds_3stages.yml --gpu 0

you can also use config file for flowers model in my github. please check this file : flowers_3stages.yml in https://github.com/rightlit/StackGAN-v2-rev/tree/main/code/cfg/

can you share the pre-trained model for flowers?

rightlit commented 3 years ago

I share you pretrained model for flowers. please refer to following link and download it.

https://drive.google.com/file/d/1-At5qsRk7IEtGKrKhAInDbQxsanRRkgE/view?usp=sharing