├── .gitignore ├── LICENSE ├── MARCONet ├── TextEnhancement.py ├── __init__.py ├── checkpoints │ ├── db_resnet34 │ │ └── 1.2 │ │ │ └── db_resnet34 │ │ │ └── readme.md │ └── download_github.py ├── models │ ├── __init__.py │ ├── networks.py │ ├── ocr.py │ ├── resnet.py │ └── textvit_arch.py └── utils │ ├── __init__.py │ ├── alphabets.py │ └── regionmerge.py ├── NOTICE ├── README.md ├── SwinIR ├── models │ └── network_swinir.py ├── utils │ └── util_calculate_psnr_ssim.py └── weights │ └── readme.md ├── assets ├── example1.png ├── example2.png ├── example3.png ├── example4.png └── example5.png ├── checkpoints └── readme.md ├── clip ├── __init__.py ├── bpe_simple_vocab_16e6.txt.gz ├── clip.py ├── model.py └── simple_tokenizer.py ├── examples ├── 1.png ├── 10.png ├── 12.png ├── 14.png ├── 3.png ├── 4.png ├── 5.png ├── 6.png ├── 7.png └── 8.png ├── gradio1.py ├── gradio2.py ├── models ├── pasd │ ├── controlnet.py │ ├── unet_2d_blocks.py │ └── unet_2d_condition.py └── pasd_light │ ├── attention.py │ ├── controlnet.py │ ├── transformer_2d.py │ ├── unet_2d_blocks.py │ └── unet_2d_condition.py ├── myutils ├── convert_lora_safetensor_to_diffusers.py ├── devices.py ├── img_util.py ├── misc.py ├── regionmerge.py ├── vaehook.py └── wavelet_color_fix.py ├── pipelines └── pipeline_pasd.py ├── realesrgan ├── __init__.py ├── archs │ ├── __init__.py │ ├── discriminator_arch.py │ └── srvgg_arch.py ├── data │ ├── __init__.py │ ├── realesrgan_dataset.py │ └── realesrgan_paired_dataset.py ├── models │ ├── __init__.py │ ├── realesrgan_model.py │ └── realesrnet_model.py ├── train.py ├── utils.py └── weights │ └── readme.md ├── requirements.txt ├── runs └── readme.md └── synthesis_vqa ├── __init__.py ├── utils ├── __init__.py ├── mos.py ├── mos_model │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ └── resnet.py │ ├── censeo_ivqa_model.py │ └── heads │ │ ├── __init__.py │ │ └── simple_head.py └── utils.py ├── vqa_cfg.json ├── vqa_model.py └── weights └── readme.md /.gitignore: -------------------------------------------------------------------------------- 1 | *.DS_Store 2 | *.pth -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/LICENSE -------------------------------------------------------------------------------- /MARCONet/TextEnhancement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/TextEnhancement.py -------------------------------------------------------------------------------- /MARCONet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MARCONet/checkpoints/db_resnet34/1.2/db_resnet34/readme.md: -------------------------------------------------------------------------------- 1 | put text detection ckpt here -------------------------------------------------------------------------------- /MARCONet/checkpoints/download_github.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/checkpoints/download_github.py -------------------------------------------------------------------------------- /MARCONet/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MARCONet/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/models/networks.py -------------------------------------------------------------------------------- /MARCONet/models/ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/models/ocr.py -------------------------------------------------------------------------------- /MARCONet/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/models/resnet.py -------------------------------------------------------------------------------- /MARCONet/models/textvit_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/models/textvit_arch.py -------------------------------------------------------------------------------- /MARCONet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MARCONet/utils/alphabets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/utils/alphabets.py -------------------------------------------------------------------------------- /MARCONet/utils/regionmerge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/MARCONet/utils/regionmerge.py -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/README.md -------------------------------------------------------------------------------- /SwinIR/models/network_swinir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/SwinIR/models/network_swinir.py -------------------------------------------------------------------------------- /SwinIR/utils/util_calculate_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/SwinIR/utils/util_calculate_psnr_ssim.py -------------------------------------------------------------------------------- /SwinIR/weights/readme.md: -------------------------------------------------------------------------------- 1 | put SwinIR checkpoints here -------------------------------------------------------------------------------- /assets/example1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/assets/example1.png -------------------------------------------------------------------------------- /assets/example2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/assets/example2.png -------------------------------------------------------------------------------- /assets/example3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/assets/example3.png -------------------------------------------------------------------------------- /assets/example4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/assets/example4.png -------------------------------------------------------------------------------- /assets/example5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/assets/example5.png -------------------------------------------------------------------------------- /checkpoints/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/checkpoints/readme.md -------------------------------------------------------------------------------- /clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/clip/clip.py -------------------------------------------------------------------------------- /clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/clip/model.py -------------------------------------------------------------------------------- /clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /examples/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/1.png -------------------------------------------------------------------------------- /examples/10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/10.png -------------------------------------------------------------------------------- /examples/12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/12.png -------------------------------------------------------------------------------- /examples/14.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/14.png -------------------------------------------------------------------------------- /examples/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/3.png -------------------------------------------------------------------------------- /examples/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/4.png -------------------------------------------------------------------------------- /examples/5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/5.png -------------------------------------------------------------------------------- /examples/6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/6.png -------------------------------------------------------------------------------- /examples/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/7.png -------------------------------------------------------------------------------- /examples/8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/examples/8.png -------------------------------------------------------------------------------- /gradio1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/gradio1.py -------------------------------------------------------------------------------- /gradio2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/gradio2.py -------------------------------------------------------------------------------- /models/pasd/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd/controlnet.py -------------------------------------------------------------------------------- /models/pasd/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd/unet_2d_blocks.py -------------------------------------------------------------------------------- /models/pasd/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd/unet_2d_condition.py -------------------------------------------------------------------------------- /models/pasd_light/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd_light/attention.py -------------------------------------------------------------------------------- /models/pasd_light/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd_light/controlnet.py -------------------------------------------------------------------------------- /models/pasd_light/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd_light/transformer_2d.py -------------------------------------------------------------------------------- /models/pasd_light/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd_light/unet_2d_blocks.py -------------------------------------------------------------------------------- /models/pasd_light/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/models/pasd_light/unet_2d_condition.py -------------------------------------------------------------------------------- /myutils/convert_lora_safetensor_to_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/myutils/convert_lora_safetensor_to_diffusers.py -------------------------------------------------------------------------------- /myutils/devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/myutils/devices.py -------------------------------------------------------------------------------- /myutils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/myutils/img_util.py -------------------------------------------------------------------------------- /myutils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/myutils/misc.py -------------------------------------------------------------------------------- /myutils/regionmerge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/myutils/regionmerge.py -------------------------------------------------------------------------------- /myutils/vaehook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/myutils/vaehook.py -------------------------------------------------------------------------------- /myutils/wavelet_color_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/myutils/wavelet_color_fix.py -------------------------------------------------------------------------------- /pipelines/pipeline_pasd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/pipelines/pipeline_pasd.py -------------------------------------------------------------------------------- /realesrgan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/__init__.py -------------------------------------------------------------------------------- /realesrgan/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/archs/__init__.py -------------------------------------------------------------------------------- /realesrgan/archs/discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/archs/discriminator_arch.py -------------------------------------------------------------------------------- /realesrgan/archs/srvgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/archs/srvgg_arch.py -------------------------------------------------------------------------------- /realesrgan/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/data/__init__.py -------------------------------------------------------------------------------- /realesrgan/data/realesrgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/data/realesrgan_dataset.py -------------------------------------------------------------------------------- /realesrgan/data/realesrgan_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/data/realesrgan_paired_dataset.py -------------------------------------------------------------------------------- /realesrgan/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/models/__init__.py -------------------------------------------------------------------------------- /realesrgan/models/realesrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/models/realesrgan_model.py -------------------------------------------------------------------------------- /realesrgan/models/realesrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/models/realesrnet_model.py -------------------------------------------------------------------------------- /realesrgan/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/train.py -------------------------------------------------------------------------------- /realesrgan/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/realesrgan/utils.py -------------------------------------------------------------------------------- /realesrgan/weights/readme.md: -------------------------------------------------------------------------------- 1 | put RealESRGAN x2 and x4 checkpoints here -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/requirements.txt -------------------------------------------------------------------------------- /runs/readme.md: -------------------------------------------------------------------------------- 1 | put trained PASD model here -------------------------------------------------------------------------------- /synthesis_vqa/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /synthesis_vqa/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /synthesis_vqa/utils/mos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/mos.py -------------------------------------------------------------------------------- /synthesis_vqa/utils/mos_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/mos_model/__init__.py -------------------------------------------------------------------------------- /synthesis_vqa/utils/mos_model/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/mos_model/backbones/__init__.py -------------------------------------------------------------------------------- /synthesis_vqa/utils/mos_model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/mos_model/backbones/resnet.py -------------------------------------------------------------------------------- /synthesis_vqa/utils/mos_model/censeo_ivqa_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/mos_model/censeo_ivqa_model.py -------------------------------------------------------------------------------- /synthesis_vqa/utils/mos_model/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/mos_model/heads/__init__.py -------------------------------------------------------------------------------- /synthesis_vqa/utils/mos_model/heads/simple_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/mos_model/heads/simple_head.py -------------------------------------------------------------------------------- /synthesis_vqa/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/utils/utils.py -------------------------------------------------------------------------------- /synthesis_vqa/vqa_cfg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/vqa_cfg.json -------------------------------------------------------------------------------- /synthesis_vqa/vqa_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dreamoving/Phantom/HEAD/synthesis_vqa/vqa_model.py -------------------------------------------------------------------------------- /synthesis_vqa/weights/readme.md: -------------------------------------------------------------------------------- 1 | put iqa checkpoints here --------------------------------------------------------------------------------