ElistratovSemyon / AI_in_II_task2

DALLE compression

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AI_in_II_task2

DALLE compression

Semantic compression algorithm realisation with DALLE2 model.

You can run it via script: main.py --config_path --data_path --gpu ,

or via main.ipynb notebook.

Config path: ./configs/dalle2.json

Average RGB-PSNR > 30.

You should download the weights and place them in the directory ./weights. Names of the weights files are registered in config path: prior.pth, decoder_1.pth and decoder_2.pth

About

DALLE compression


Languages

Language:Jupyter Notebook 99.1%Language:Python 0.9%