├── .gitignore ├── LICENSE ├── README.md ├── bert_tokenizer.py ├── cldm ├── cldm.py ├── ddim_hacked.py ├── embedding_manager.py ├── hack.py ├── logger.py ├── model.py └── recognizer.py ├── dataset_util.py ├── demo.py ├── docs ├── demo.jpg ├── eval.jpg ├── framework.jpg ├── gallery.png ├── reproduce.jpg └── sample.jpg ├── environment.yaml ├── eval ├── anytext_multiGPUs.py ├── anytext_singleGPU.py ├── controlnet_multiGPUs.py ├── controlnet_singleGPU.py ├── eval_dgocr.py ├── eval_fid.sh ├── eval_ocr.sh ├── gen_glyph.sh ├── gen_imgs_anytext.sh ├── gen_imgs_controlnet_canny.sh ├── gen_imgs_glyphcontrol.sh ├── gen_imgs_textdiffuser.sh ├── glyphcontrol_multiGPUs.py ├── glyphcontrol_singleGPU.py ├── render_glyph_imgs.py ├── textdiffuser_multiGPUs.py └── textdiffuser_singleGPU.py ├── example_images ├── banner.png ├── edit1.png ├── edit10.png ├── edit11.png ├── edit12.png ├── edit13.png ├── edit14.png ├── edit15.png ├── edit16.png ├── edit2.png ├── edit3.png ├── edit4.png ├── edit5.png ├── edit6.png ├── edit7.png ├── edit8.png ├── edit9.png ├── gen1.png ├── gen10.png ├── gen11.png ├── gen12.png ├── gen13.png ├── gen14.png ├── gen15.png ├── gen16.png ├── gen17.png ├── gen18.png ├── gen19.png ├── gen2.png ├── gen20.png ├── gen21.png ├── gen3.png ├── gen4.png ├── gen5.png ├── gen6.png ├── gen7.png ├── gen8.png ├── gen9.png ├── ref1.jpg ├── ref10.jpg ├── ref11.jpg ├── ref12.png ├── ref13.jpg ├── ref14.png ├── ref15.jpeg ├── ref16.jpeg ├── ref2.jpg ├── ref3.jpg ├── ref4.jpg ├── ref5.jpg ├── ref6.jpg ├── ref7.jpg ├── ref8.jpg └── ref9.jpg ├── font └── put_your_font_file_here.txt ├── inference.py ├── javascript └── bboxHint.js ├── ldm ├── data │ ├── __init__.py │ └── util.py ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── dpm_solver │ │ ├── __init__.py │ │ ├── dpm_solver.py │ │ └── sampler.py │ │ ├── plms.py │ │ └── sampling_util.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ ├── upscaling.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ └── modules.py │ ├── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ ├── utils │ │ │ └── test.png │ │ └── utils_image.py │ └── midas │ │ ├── __init__.py │ │ ├── api.py │ │ ├── midas │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── blocks.py │ │ ├── dpt_depth.py │ │ ├── midas_net.py │ │ ├── midas_net_custom.py │ │ ├── transforms.py │ │ └── vit.py │ │ └── utils.py └── util.py ├── lora_util.py ├── models_yaml ├── anytext_sd15.yaml ├── anytext_sd15_conv.yaml ├── anytext_sd15_perloss.yaml └── anytext_sd15_vit.yaml ├── ocr_recog ├── RNN.py ├── RecCTCHead.py ├── RecModel.py ├── RecMv1_enhance.py ├── RecSVTR.py ├── common.py ├── en_dict.txt └── ppocr_keys_v1.txt ├── ocr_weights ├── en_dict.txt ├── ppocr_keys_v1.txt ├── ppv3_rec.pth └── ppv3_rec_en.pth ├── style.css ├── t3_dataset.py ├── tool_add_anytext.py ├── train.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/README.md -------------------------------------------------------------------------------- /bert_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/bert_tokenizer.py -------------------------------------------------------------------------------- /cldm/cldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/cldm/cldm.py -------------------------------------------------------------------------------- /cldm/ddim_hacked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/cldm/ddim_hacked.py -------------------------------------------------------------------------------- /cldm/embedding_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/cldm/embedding_manager.py -------------------------------------------------------------------------------- /cldm/hack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/cldm/hack.py -------------------------------------------------------------------------------- /cldm/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/cldm/logger.py -------------------------------------------------------------------------------- /cldm/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/cldm/model.py -------------------------------------------------------------------------------- /cldm/recognizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/cldm/recognizer.py -------------------------------------------------------------------------------- /dataset_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/dataset_util.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/demo.py -------------------------------------------------------------------------------- /docs/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/docs/demo.jpg -------------------------------------------------------------------------------- /docs/eval.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/docs/eval.jpg -------------------------------------------------------------------------------- /docs/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/docs/framework.jpg -------------------------------------------------------------------------------- /docs/gallery.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/docs/gallery.png -------------------------------------------------------------------------------- /docs/reproduce.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/docs/reproduce.jpg -------------------------------------------------------------------------------- /docs/sample.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/docs/sample.jpg -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/environment.yaml -------------------------------------------------------------------------------- /eval/anytext_multiGPUs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/anytext_multiGPUs.py -------------------------------------------------------------------------------- /eval/anytext_singleGPU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/anytext_singleGPU.py -------------------------------------------------------------------------------- /eval/controlnet_multiGPUs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/controlnet_multiGPUs.py -------------------------------------------------------------------------------- /eval/controlnet_singleGPU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/controlnet_singleGPU.py -------------------------------------------------------------------------------- /eval/eval_dgocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/eval_dgocr.py -------------------------------------------------------------------------------- /eval/eval_fid.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/eval_fid.sh -------------------------------------------------------------------------------- /eval/eval_ocr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/eval_ocr.sh -------------------------------------------------------------------------------- /eval/gen_glyph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/gen_glyph.sh -------------------------------------------------------------------------------- /eval/gen_imgs_anytext.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/gen_imgs_anytext.sh -------------------------------------------------------------------------------- /eval/gen_imgs_controlnet_canny.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/gen_imgs_controlnet_canny.sh -------------------------------------------------------------------------------- /eval/gen_imgs_glyphcontrol.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/gen_imgs_glyphcontrol.sh -------------------------------------------------------------------------------- /eval/gen_imgs_textdiffuser.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/gen_imgs_textdiffuser.sh -------------------------------------------------------------------------------- /eval/glyphcontrol_multiGPUs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/glyphcontrol_multiGPUs.py -------------------------------------------------------------------------------- /eval/glyphcontrol_singleGPU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/glyphcontrol_singleGPU.py -------------------------------------------------------------------------------- /eval/render_glyph_imgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/render_glyph_imgs.py -------------------------------------------------------------------------------- /eval/textdiffuser_multiGPUs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/textdiffuser_multiGPUs.py -------------------------------------------------------------------------------- /eval/textdiffuser_singleGPU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/eval/textdiffuser_singleGPU.py -------------------------------------------------------------------------------- /example_images/banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/banner.png -------------------------------------------------------------------------------- /example_images/edit1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit1.png -------------------------------------------------------------------------------- /example_images/edit10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit10.png -------------------------------------------------------------------------------- /example_images/edit11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit11.png -------------------------------------------------------------------------------- /example_images/edit12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit12.png -------------------------------------------------------------------------------- /example_images/edit13.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit13.png -------------------------------------------------------------------------------- /example_images/edit14.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit14.png -------------------------------------------------------------------------------- /example_images/edit15.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit15.png -------------------------------------------------------------------------------- /example_images/edit16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit16.png -------------------------------------------------------------------------------- /example_images/edit2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit2.png -------------------------------------------------------------------------------- /example_images/edit3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit3.png -------------------------------------------------------------------------------- /example_images/edit4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit4.png -------------------------------------------------------------------------------- /example_images/edit5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit5.png -------------------------------------------------------------------------------- /example_images/edit6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit6.png -------------------------------------------------------------------------------- /example_images/edit7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit7.png -------------------------------------------------------------------------------- /example_images/edit8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit8.png -------------------------------------------------------------------------------- /example_images/edit9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/edit9.png -------------------------------------------------------------------------------- /example_images/gen1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen1.png -------------------------------------------------------------------------------- /example_images/gen10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen10.png -------------------------------------------------------------------------------- /example_images/gen11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen11.png -------------------------------------------------------------------------------- /example_images/gen12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen12.png -------------------------------------------------------------------------------- /example_images/gen13.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen13.png -------------------------------------------------------------------------------- /example_images/gen14.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen14.png -------------------------------------------------------------------------------- /example_images/gen15.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen15.png -------------------------------------------------------------------------------- /example_images/gen16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen16.png -------------------------------------------------------------------------------- /example_images/gen17.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen17.png -------------------------------------------------------------------------------- /example_images/gen18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen18.png -------------------------------------------------------------------------------- /example_images/gen19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen19.png -------------------------------------------------------------------------------- /example_images/gen2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen2.png -------------------------------------------------------------------------------- /example_images/gen20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen20.png -------------------------------------------------------------------------------- /example_images/gen21.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen21.png -------------------------------------------------------------------------------- /example_images/gen3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen3.png -------------------------------------------------------------------------------- /example_images/gen4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen4.png -------------------------------------------------------------------------------- /example_images/gen5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen5.png -------------------------------------------------------------------------------- /example_images/gen6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen6.png -------------------------------------------------------------------------------- /example_images/gen7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen7.png -------------------------------------------------------------------------------- /example_images/gen8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen8.png -------------------------------------------------------------------------------- /example_images/gen9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/gen9.png -------------------------------------------------------------------------------- /example_images/ref1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref1.jpg -------------------------------------------------------------------------------- /example_images/ref10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref10.jpg -------------------------------------------------------------------------------- /example_images/ref11.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref11.jpg -------------------------------------------------------------------------------- /example_images/ref12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref12.png -------------------------------------------------------------------------------- /example_images/ref13.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref13.jpg -------------------------------------------------------------------------------- /example_images/ref14.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref14.png -------------------------------------------------------------------------------- /example_images/ref15.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref15.jpeg -------------------------------------------------------------------------------- /example_images/ref16.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref16.jpeg -------------------------------------------------------------------------------- /example_images/ref2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref2.jpg -------------------------------------------------------------------------------- /example_images/ref3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref3.jpg -------------------------------------------------------------------------------- /example_images/ref4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref4.jpg -------------------------------------------------------------------------------- /example_images/ref5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref5.jpg -------------------------------------------------------------------------------- /example_images/ref6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref6.jpg -------------------------------------------------------------------------------- /example_images/ref7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref7.jpg -------------------------------------------------------------------------------- /example_images/ref8.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref8.jpg -------------------------------------------------------------------------------- /example_images/ref9.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/example_images/ref9.jpg -------------------------------------------------------------------------------- /font/put_your_font_file_here.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/inference.py -------------------------------------------------------------------------------- /javascript/bboxHint.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/javascript/bboxHint.js -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/data/util.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/models/diffusion/sampling_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/models/diffusion/sampling_util.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/api.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/midas/base_model.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/midas/blocks.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/midas/dpt_depth.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/midas/midas_net.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/midas/midas_net_custom.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/midas/transforms.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/midas/vit.py -------------------------------------------------------------------------------- /ldm/modules/midas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/modules/midas/utils.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ldm/util.py -------------------------------------------------------------------------------- /lora_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/lora_util.py -------------------------------------------------------------------------------- /models_yaml/anytext_sd15.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/models_yaml/anytext_sd15.yaml -------------------------------------------------------------------------------- /models_yaml/anytext_sd15_conv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/models_yaml/anytext_sd15_conv.yaml -------------------------------------------------------------------------------- /models_yaml/anytext_sd15_perloss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/models_yaml/anytext_sd15_perloss.yaml -------------------------------------------------------------------------------- /models_yaml/anytext_sd15_vit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/models_yaml/anytext_sd15_vit.yaml -------------------------------------------------------------------------------- /ocr_recog/RNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/RNN.py -------------------------------------------------------------------------------- /ocr_recog/RecCTCHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/RecCTCHead.py -------------------------------------------------------------------------------- /ocr_recog/RecModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/RecModel.py -------------------------------------------------------------------------------- /ocr_recog/RecMv1_enhance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/RecMv1_enhance.py -------------------------------------------------------------------------------- /ocr_recog/RecSVTR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/RecSVTR.py -------------------------------------------------------------------------------- /ocr_recog/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/common.py -------------------------------------------------------------------------------- /ocr_recog/en_dict.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/en_dict.txt -------------------------------------------------------------------------------- /ocr_recog/ppocr_keys_v1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_recog/ppocr_keys_v1.txt -------------------------------------------------------------------------------- /ocr_weights/en_dict.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_weights/en_dict.txt -------------------------------------------------------------------------------- /ocr_weights/ppocr_keys_v1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_weights/ppocr_keys_v1.txt -------------------------------------------------------------------------------- /ocr_weights/ppv3_rec.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_weights/ppv3_rec.pth -------------------------------------------------------------------------------- /ocr_weights/ppv3_rec_en.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/ocr_weights/ppv3_rec_en.pth -------------------------------------------------------------------------------- /style.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/style.css -------------------------------------------------------------------------------- /t3_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/t3_dataset.py -------------------------------------------------------------------------------- /tool_add_anytext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/tool_add_anytext.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/train.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tyxsspa/AnyText/HEAD/util.py --------------------------------------------------------------------------------