search

fudan-zvg / GSS

[CVPR 2023] Official repository of Generative Semantic Segmentation
204 stars 12 forks source link

Could you share the weights of segmentation head? #3

Closed AugustLHHHHHH closed 1 year ago

AugustLHHHHHH commented 1 year ago

Hi~ I try to run tools/pytorch2onnx.py and then return error: " No such file or directory: /home/chenjiaqi/pj/mmsegmentation/ckp/encoder.pkl". So could you share the weight? Thanks!

Jiaqi-Chen-00 commented 1 year ago

As stated in the paper, we utilize the DALL-E pre-trained VQVAE weights and freeze both the encoder and decoder. Please download the encoder and decoder weights at https://cdn.openai.com/dall-e/encoder.pkl and https://cdn.openai.com/dall-e/decoder.pkl

AugustLHHHHHH commented 1 year ago

The download link is available, thank you very much for your sharing!