├── README.md ├── dataset.py ├── environment.yml ├── eval_rl.py ├── ic_utils.py ├── images └── teaser.png ├── module.py ├── prep ├── prep_coco.py └── prep_coco_ru.py ├── train.py ├── train_rl.py ├── train_ru.py └── train_ru_rl.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/README.md -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/dataset.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/environment.yml -------------------------------------------------------------------------------- /eval_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/eval_rl.py -------------------------------------------------------------------------------- /ic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/ic_utils.py -------------------------------------------------------------------------------- /images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/images/teaser.png -------------------------------------------------------------------------------- /module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/module.py -------------------------------------------------------------------------------- /prep/prep_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/prep/prep_coco.py -------------------------------------------------------------------------------- /prep/prep_coco_ru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/prep/prep_coco_ru.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/train.py -------------------------------------------------------------------------------- /train_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/train_rl.py -------------------------------------------------------------------------------- /train_ru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/train_ru.py -------------------------------------------------------------------------------- /train_ru_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zipengxuc/StylerDALLE/HEAD/train_ru_rl.py --------------------------------------------------------------------------------