mage icon indicating copy to clipboard operation
mage copied to clipboard

Question about applying to my own dataset

Open mabaisen opened this issue 2 years ago • 2 comments

Hello, i want to use my own galaxy dataset, which is completely different from ImageNet. Should I retrain VQGAN from scratch.

J000000 80+004200 0 256

mabaisen avatar Dec 10 '23 02:12 mabaisen

Yes you need to re-train the VQGAN -- the provided VQGAN checkpoint is pre-trained on ImageNet-1K, which does not include galaxy images. Unfortunately, we cannot release the code for VQGAN training, but the original VQGAN code should give reasonable performance.

LTH14 avatar Dec 10 '23 03:12 LTH14

Yes you need to re-train the VQGAN -- the provided VQGAN checkpoint is pre-trained on ImageNet-1K, which does not include galaxy images. Unfortunately, we cannot release the code for VQGAN training, but the original VQGAN code should give reasonable performance.

Ok, thank you very much for your answer.

mabaisen avatar Dec 10 '23 06:12 mabaisen