├── .idea ├── .gitignore ├── CT2.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── README.md ├── application.pdf ├── application.png ├── clean_train_filenames.zip ├── mask_prior.pickle ├── q_prior_all ├── L1-all.npy ├── L2-all.npy ├── L3-all.npy └── L4-all.npy ├── requirements.txt ├── segm ├── config.py ├── config.yml ├── data │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── coco.cpython-36.pyc │ │ ├── factory.cpython-36.pyc │ │ ├── loader.cpython-36.pyc │ │ ├── random.cpython-36.pyc │ │ └── utils.cpython-36.pyc │ ├── ade20k.py │ ├── base.py │ ├── cityscapes.py │ ├── coco.py │ ├── config │ │ ├── ade20k.py │ │ ├── ade20k.yml │ │ ├── cityscapes.py │ │ ├── cityscapes.yml │ │ ├── pascal_context.py │ │ └── pascal_context.yml │ ├── factory.py │ ├── loader.py │ ├── pascal_context.py │ ├── random.py │ └── utils.py ├── engine.py ├── eval │ ├── accuracy.py │ └── miou.py ├── eval_colorfullness.py ├── eval_fid.py ├── eval_metrics.py ├── make_q_actual.py ├── metrics.py ├── model │ ├── __pycache__ │ │ ├── blocks.cpython-36.pyc │ │ ├── decoder.cpython-36.pyc │ │ ├── factory.cpython-36.pyc │ │ ├── segmenter.cpython-36.pyc │ │ ├── utils.cpython-36.pyc │ │ ├── utils.cpython-37.pyc │ │ └── vit.cpython-36.pyc │ ├── blocks.py │ ├── decoder.py │ ├── decoder_ori.py │ ├── factory.py │ ├── segmenter.py │ ├── utils.py │ └── vit.py ├── optim │ ├── __pycache__ │ │ ├── factory.cpython-36.pyc │ │ └── scheduler.cpython-36.pyc │ ├── factory.py │ └── scheduler.py ├── resources │ ├── ab-gamut.npy │ └── q-prior.npy ├── single_test.py ├── test.py ├── train.py └── utils │ ├── __pycache__ │ ├── distributed.cpython-36.pyc │ ├── logger.cpython-36.pyc │ └── torch.cpython-36.pyc │ ├── distributed.py │ ├── download.py │ ├── lines.py │ ├── logger.py │ ├── logs.py │ └── torch.py └── setup.py /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /workspace.xml -------------------------------------------------------------------------------- /.idea/CT2.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/.idea/CT2.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/README.md -------------------------------------------------------------------------------- /application.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/application.pdf -------------------------------------------------------------------------------- /application.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/application.png -------------------------------------------------------------------------------- /clean_train_filenames.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/clean_train_filenames.zip -------------------------------------------------------------------------------- /mask_prior.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/mask_prior.pickle -------------------------------------------------------------------------------- /q_prior_all/L1-all.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/q_prior_all/L1-all.npy -------------------------------------------------------------------------------- /q_prior_all/L2-all.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/q_prior_all/L2-all.npy -------------------------------------------------------------------------------- /q_prior_all/L3-all.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/q_prior_all/L3-all.npy -------------------------------------------------------------------------------- /q_prior_all/L4-all.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/q_prior_all/L4-all.npy -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/requirements.txt -------------------------------------------------------------------------------- /segm/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/config.py -------------------------------------------------------------------------------- /segm/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/config.yml -------------------------------------------------------------------------------- /segm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/__init__.py -------------------------------------------------------------------------------- /segm/data/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /segm/data/__pycache__/coco.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/__pycache__/coco.cpython-36.pyc -------------------------------------------------------------------------------- /segm/data/__pycache__/factory.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/__pycache__/factory.cpython-36.pyc -------------------------------------------------------------------------------- /segm/data/__pycache__/loader.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/__pycache__/loader.cpython-36.pyc -------------------------------------------------------------------------------- /segm/data/__pycache__/random.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/__pycache__/random.cpython-36.pyc -------------------------------------------------------------------------------- /segm/data/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /segm/data/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/ade20k.py -------------------------------------------------------------------------------- /segm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/base.py -------------------------------------------------------------------------------- /segm/data/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/cityscapes.py -------------------------------------------------------------------------------- /segm/data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/coco.py -------------------------------------------------------------------------------- /segm/data/config/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/config/ade20k.py -------------------------------------------------------------------------------- /segm/data/config/ade20k.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/config/ade20k.yml -------------------------------------------------------------------------------- /segm/data/config/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/config/cityscapes.py -------------------------------------------------------------------------------- /segm/data/config/cityscapes.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/config/cityscapes.yml -------------------------------------------------------------------------------- /segm/data/config/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/config/pascal_context.py -------------------------------------------------------------------------------- /segm/data/config/pascal_context.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/config/pascal_context.yml -------------------------------------------------------------------------------- /segm/data/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/factory.py -------------------------------------------------------------------------------- /segm/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/loader.py -------------------------------------------------------------------------------- /segm/data/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/pascal_context.py -------------------------------------------------------------------------------- /segm/data/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/random.py -------------------------------------------------------------------------------- /segm/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/data/utils.py -------------------------------------------------------------------------------- /segm/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/engine.py -------------------------------------------------------------------------------- /segm/eval/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/eval/accuracy.py -------------------------------------------------------------------------------- /segm/eval/miou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/eval/miou.py -------------------------------------------------------------------------------- /segm/eval_colorfullness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/eval_colorfullness.py -------------------------------------------------------------------------------- /segm/eval_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/eval_fid.py -------------------------------------------------------------------------------- /segm/eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/eval_metrics.py -------------------------------------------------------------------------------- /segm/make_q_actual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/make_q_actual.py -------------------------------------------------------------------------------- /segm/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/metrics.py -------------------------------------------------------------------------------- /segm/model/__pycache__/blocks.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/__pycache__/blocks.cpython-36.pyc -------------------------------------------------------------------------------- /segm/model/__pycache__/decoder.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/__pycache__/decoder.cpython-36.pyc -------------------------------------------------------------------------------- /segm/model/__pycache__/factory.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/__pycache__/factory.cpython-36.pyc -------------------------------------------------------------------------------- /segm/model/__pycache__/segmenter.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/__pycache__/segmenter.cpython-36.pyc -------------------------------------------------------------------------------- /segm/model/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /segm/model/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /segm/model/__pycache__/vit.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/__pycache__/vit.cpython-36.pyc -------------------------------------------------------------------------------- /segm/model/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/blocks.py -------------------------------------------------------------------------------- /segm/model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/decoder.py -------------------------------------------------------------------------------- /segm/model/decoder_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/decoder_ori.py -------------------------------------------------------------------------------- /segm/model/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/factory.py -------------------------------------------------------------------------------- /segm/model/segmenter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/segmenter.py -------------------------------------------------------------------------------- /segm/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/utils.py -------------------------------------------------------------------------------- /segm/model/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/model/vit.py -------------------------------------------------------------------------------- /segm/optim/__pycache__/factory.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/optim/__pycache__/factory.cpython-36.pyc -------------------------------------------------------------------------------- /segm/optim/__pycache__/scheduler.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/optim/__pycache__/scheduler.cpython-36.pyc -------------------------------------------------------------------------------- /segm/optim/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/optim/factory.py -------------------------------------------------------------------------------- /segm/optim/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/optim/scheduler.py -------------------------------------------------------------------------------- /segm/resources/ab-gamut.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/resources/ab-gamut.npy -------------------------------------------------------------------------------- /segm/resources/q-prior.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/resources/q-prior.npy -------------------------------------------------------------------------------- /segm/single_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/single_test.py -------------------------------------------------------------------------------- /segm/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/test.py -------------------------------------------------------------------------------- /segm/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/train.py -------------------------------------------------------------------------------- /segm/utils/__pycache__/distributed.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/__pycache__/distributed.cpython-36.pyc -------------------------------------------------------------------------------- /segm/utils/__pycache__/logger.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/__pycache__/logger.cpython-36.pyc -------------------------------------------------------------------------------- /segm/utils/__pycache__/torch.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/__pycache__/torch.cpython-36.pyc -------------------------------------------------------------------------------- /segm/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/distributed.py -------------------------------------------------------------------------------- /segm/utils/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/download.py -------------------------------------------------------------------------------- /segm/utils/lines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/lines.py -------------------------------------------------------------------------------- /segm/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/logger.py -------------------------------------------------------------------------------- /segm/utils/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/logs.py -------------------------------------------------------------------------------- /segm/utils/torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/segm/utils/torch.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuchenweng/CT2/HEAD/setup.py --------------------------------------------------------------------------------