├── README.md ├── configs └── X2Mari.yaml ├── environment.yml ├── evalution ├── 1.0_cal_gt.sh ├── 1.1_cal_fid.sh ├── 2.0_crop_box.py ├── 2.1_cal_cas.py ├── 3.0_pro_labels.py └── 3.1_cal_yolos.py ├── img_file ├── abstract.jpg ├── generation.jpg ├── labelling.jpg ├── logo_Neptune-X.png └── selection.jpg ├── inference.py ├── ldm ├── data │ ├── base.py │ ├── dataset.py │ ├── dataset_test.py │ └── magicship_tokinzer_input.py ├── lr_scheduler.py ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── ldm.py │ │ └── plms.py ├── modules │ ├── attention.py │ ├── convert_ckpt.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── convnext.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ ├── pseudo_example.py │ │ ├── resnet.py │ │ ├── text_grounding_net.py │ │ └── util.py │ ├── distributed.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ ├── modules.py │ │ └── modules_backup.py │ ├── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ └── utils_image.py │ ├── losses │ │ ├── __init__.py │ │ ├── attn_loss.py │ │ ├── contperceptual.py │ │ └── vqperceptual.py │ └── x_transformer.py └── util.py ├── sample ├── 1.json └── 2.json ├── test.py ├── train.py └── trainer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/README.md -------------------------------------------------------------------------------- /configs/X2Mari.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/configs/X2Mari.yaml -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/environment.yml -------------------------------------------------------------------------------- /evalution/1.0_cal_gt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/evalution/1.0_cal_gt.sh -------------------------------------------------------------------------------- /evalution/1.1_cal_fid.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/evalution/1.1_cal_fid.sh -------------------------------------------------------------------------------- /evalution/2.0_crop_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/evalution/2.0_crop_box.py -------------------------------------------------------------------------------- /evalution/2.1_cal_cas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/evalution/2.1_cal_cas.py -------------------------------------------------------------------------------- /evalution/3.0_pro_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/evalution/3.0_pro_labels.py -------------------------------------------------------------------------------- /evalution/3.1_cal_yolos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/evalution/3.1_cal_yolos.py -------------------------------------------------------------------------------- /img_file/abstract.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/img_file/abstract.jpg -------------------------------------------------------------------------------- /img_file/generation.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/img_file/generation.jpg -------------------------------------------------------------------------------- /img_file/labelling.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/img_file/labelling.jpg -------------------------------------------------------------------------------- /img_file/logo_Neptune-X.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/img_file/logo_Neptune-X.png -------------------------------------------------------------------------------- /img_file/selection.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/img_file/selection.jpg -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/inference.py -------------------------------------------------------------------------------- /ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/data/base.py -------------------------------------------------------------------------------- /ldm/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/data/dataset.py -------------------------------------------------------------------------------- /ldm/data/dataset_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/data/dataset_test.py -------------------------------------------------------------------------------- /ldm/data/magicship_tokinzer_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/data/magicship_tokinzer_input.py -------------------------------------------------------------------------------- /ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/models/diffusion/ldm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/convert_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/convert_ckpt.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/diffusionmodules/convnext.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/pseudo_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/diffusionmodules/pseudo_example.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/diffusionmodules/resnet.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/text_grounding_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/diffusionmodules/text_grounding_net.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/distributed.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/encoders/modules_backup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/encoders/modules_backup.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /ldm/modules/losses/attn_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/losses/attn_loss.py -------------------------------------------------------------------------------- /ldm/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/ldm/util.py -------------------------------------------------------------------------------- /sample/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/sample/1.json -------------------------------------------------------------------------------- /sample/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/sample/2.json -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gy65896/Neptune-X/HEAD/trainer.py --------------------------------------------------------------------------------