├── .gitignore ├── README.md ├── beyond ├── enhancement │ ├── .gitignore │ ├── README.md │ ├── data │ │ ├── __init__.py │ │ ├── adobe5k_dataset.py │ │ ├── base_dataset.py │ │ ├── image_folder.py │ │ └── loader.py │ ├── models │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── base_networks.py │ │ ├── harmony_networks.py │ │ ├── ht_model.py │ │ ├── position_encoding.py │ │ └── transformer.py │ ├── options │ │ ├── __init__.py │ │ ├── base_options.py │ │ ├── test_options.py │ │ └── train_options.py │ ├── test.py │ ├── train.py │ ├── train_net.py │ └── util │ │ ├── distributed.py │ │ ├── evaluation.py │ │ ├── flops.py │ │ ├── html.py │ │ ├── loss.py │ │ ├── misc copy.py │ │ ├── misc.py │ │ ├── multiprocessing.py │ │ ├── ssim.py │ │ ├── tools.py │ │ ├── util.py │ │ └── visualizer.py ├── inpainting │ ├── README.md │ ├── checkpoints │ │ └── paris-fcin-deconvout-2H6L │ │ │ └── config.yml │ ├── dataset │ │ └── paris │ │ │ ├── testmask.flist │ │ │ ├── train.flist │ │ │ └── val.flist │ ├── main.py │ ├── scores │ │ ├── __pycache__ │ │ │ └── inception.cpython-37.pyc │ │ ├── fid.py │ │ ├── fid_score.py │ │ ├── inception.py │ │ ├── metrics │ │ └── metrics.py │ ├── scripts │ │ ├── cam.py │ │ ├── cramp_img.py │ │ ├── flist.py │ │ ├── flower_resize.py │ │ ├── norm.py │ │ ├── printvgg19.py │ │ └── test_rcblayer.py │ ├── src │ │ ├── INN.py │ │ ├── ImagineGAN.py │ │ ├── config.py │ │ ├── dataset.py │ │ ├── masks.py │ │ ├── metrics.py │ │ ├── model │ │ │ ├── atten.py │ │ │ ├── loss.py │ │ │ ├── model.py │ │ │ ├── networks.py │ │ │ ├── partialconv.py │ │ │ ├── rcb_layer.py │ │ │ ├── reshape.py │ │ │ └── transformer.py │ │ ├── tools.py │ │ ├── utils.py │ │ └── utilsgzh.py │ ├── test.py │ └── train.py ├── relighting │ ├── .gitignore │ ├── README.md │ ├── data.py │ ├── data │ │ ├── __init__.py │ │ ├── base_dataset.py │ │ ├── dpr_dataset.py │ │ ├── dprtransfer_dataset.py │ │ ├── image_folder.py │ │ └── loader.py │ ├── datasets │ │ ├── randomdpr_test.txt │ │ ├── randomdprtgt_test.txt │ │ └── train.lst │ ├── models │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── base_networks.py │ │ ├── dht_model.py │ │ ├── harmony_networks.py │ │ ├── position_encoding.py │ │ └── transformer.py │ ├── options │ │ ├── __init__.py │ │ ├── base_options.py │ │ ├── test_options.py │ │ └── train_options.py │ ├── test.py │ ├── train.py │ ├── train_net.py │ └── util │ │ ├── distributed.py │ │ ├── evaluation.py │ │ ├── flops.py │ │ ├── html.py │ │ ├── loss.py │ │ ├── misc copy.py │ │ ├── misc.py │ │ ├── multiprocessing.py │ │ ├── ssim.py │ │ ├── tools.py │ │ ├── util.py │ │ └── visualizer.py └── white_balance │ ├── README.md │ ├── data │ ├── __init__.py │ ├── base_dataset.py │ ├── folds │ │ ├── fold1.mat │ │ ├── fold1_.mat │ │ ├── fold2.mat │ │ ├── fold2_.mat │ │ ├── fold3.mat │ │ └── fold3_.mat │ ├── image_folder.py │ ├── loader.py │ ├── set1test_dataset.py │ ├── set_1_test.txt │ ├── train.txt │ └── wb_dataset.py │ ├── models │ ├── __init__.py │ ├── base_model.py │ ├── base_networks.py │ ├── harmony_networks.py │ ├── ht_model.py │ ├── position_encoding.py │ └── transformer.py │ ├── options │ ├── __init__.py │ ├── base_options.py │ ├── test_options.py │ └── train_options.py │ ├── test.py │ ├── train.py │ ├── train_net.py │ └── util │ ├── distributed.py │ ├── evaluation.py │ ├── flops.py │ ├── html.py │ ├── loss.py │ ├── misc copy.py │ ├── misc.py │ ├── multiprocessing.py │ ├── ssim.py │ ├── tools.py │ ├── util.py │ ├── utils.py │ └── visualizer.py ├── data ├── __init__.py ├── base_dataset.py ├── ihd_dataset.py ├── image_folder.py ├── loader.py └── real_dataset.py ├── evaluation ├── ih_evaluation.py └── pytorch_ssim.py ├── models ├── __init__.py ├── base_model.py ├── base_networks.py ├── cnnht_model.py ├── dht_model.py ├── harmony_networks.py ├── ht_model.py ├── position_encoding.py └── transformer.py ├── options ├── __init__.py ├── base_options.py ├── test_options.py └── train_options.py ├── test.py ├── train.py ├── train_net.py └── util ├── distributed.py ├── evaluation.py ├── flops.py ├── html.py ├── loss.py ├── misc copy.py ├── misc.py ├── multiprocessing.py ├── ssim.py ├── tools.py ├── util.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/README.md -------------------------------------------------------------------------------- /beyond/enhancement/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/.gitignore -------------------------------------------------------------------------------- /beyond/enhancement/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/README.md -------------------------------------------------------------------------------- /beyond/enhancement/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/data/__init__.py -------------------------------------------------------------------------------- /beyond/enhancement/data/adobe5k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/data/adobe5k_dataset.py -------------------------------------------------------------------------------- /beyond/enhancement/data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/data/base_dataset.py -------------------------------------------------------------------------------- /beyond/enhancement/data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/data/image_folder.py -------------------------------------------------------------------------------- /beyond/enhancement/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/data/loader.py -------------------------------------------------------------------------------- /beyond/enhancement/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/models/__init__.py -------------------------------------------------------------------------------- /beyond/enhancement/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/models/base_model.py -------------------------------------------------------------------------------- /beyond/enhancement/models/base_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/models/base_networks.py -------------------------------------------------------------------------------- /beyond/enhancement/models/harmony_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/models/harmony_networks.py -------------------------------------------------------------------------------- /beyond/enhancement/models/ht_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/models/ht_model.py -------------------------------------------------------------------------------- /beyond/enhancement/models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/models/position_encoding.py -------------------------------------------------------------------------------- /beyond/enhancement/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/models/transformer.py -------------------------------------------------------------------------------- /beyond/enhancement/options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/options/__init__.py -------------------------------------------------------------------------------- /beyond/enhancement/options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/options/base_options.py -------------------------------------------------------------------------------- /beyond/enhancement/options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/options/test_options.py -------------------------------------------------------------------------------- /beyond/enhancement/options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/options/train_options.py -------------------------------------------------------------------------------- /beyond/enhancement/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/test.py -------------------------------------------------------------------------------- /beyond/enhancement/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/train.py -------------------------------------------------------------------------------- /beyond/enhancement/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/train_net.py -------------------------------------------------------------------------------- /beyond/enhancement/util/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/distributed.py -------------------------------------------------------------------------------- /beyond/enhancement/util/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/evaluation.py -------------------------------------------------------------------------------- /beyond/enhancement/util/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/flops.py -------------------------------------------------------------------------------- /beyond/enhancement/util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/html.py -------------------------------------------------------------------------------- /beyond/enhancement/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/loss.py -------------------------------------------------------------------------------- /beyond/enhancement/util/misc copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/misc copy.py -------------------------------------------------------------------------------- /beyond/enhancement/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/misc.py -------------------------------------------------------------------------------- /beyond/enhancement/util/multiprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/multiprocessing.py -------------------------------------------------------------------------------- /beyond/enhancement/util/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/ssim.py -------------------------------------------------------------------------------- /beyond/enhancement/util/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/tools.py -------------------------------------------------------------------------------- /beyond/enhancement/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/util.py -------------------------------------------------------------------------------- /beyond/enhancement/util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/enhancement/util/visualizer.py -------------------------------------------------------------------------------- /beyond/inpainting/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/README.md -------------------------------------------------------------------------------- /beyond/inpainting/checkpoints/paris-fcin-deconvout-2H6L/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/checkpoints/paris-fcin-deconvout-2H6L/config.yml -------------------------------------------------------------------------------- /beyond/inpainting/dataset/paris/testmask.flist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/dataset/paris/testmask.flist -------------------------------------------------------------------------------- /beyond/inpainting/dataset/paris/train.flist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/dataset/paris/train.flist -------------------------------------------------------------------------------- /beyond/inpainting/dataset/paris/val.flist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/dataset/paris/val.flist -------------------------------------------------------------------------------- /beyond/inpainting/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/main.py -------------------------------------------------------------------------------- /beyond/inpainting/scores/__pycache__/inception.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scores/__pycache__/inception.cpython-37.pyc -------------------------------------------------------------------------------- /beyond/inpainting/scores/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scores/fid.py -------------------------------------------------------------------------------- /beyond/inpainting/scores/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scores/fid_score.py -------------------------------------------------------------------------------- /beyond/inpainting/scores/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scores/inception.py -------------------------------------------------------------------------------- /beyond/inpainting/scores/metrics: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scores/metrics -------------------------------------------------------------------------------- /beyond/inpainting/scores/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scores/metrics.py -------------------------------------------------------------------------------- /beyond/inpainting/scripts/cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scripts/cam.py -------------------------------------------------------------------------------- /beyond/inpainting/scripts/cramp_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scripts/cramp_img.py -------------------------------------------------------------------------------- /beyond/inpainting/scripts/flist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scripts/flist.py -------------------------------------------------------------------------------- /beyond/inpainting/scripts/flower_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scripts/flower_resize.py -------------------------------------------------------------------------------- /beyond/inpainting/scripts/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scripts/norm.py -------------------------------------------------------------------------------- /beyond/inpainting/scripts/printvgg19.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scripts/printvgg19.py -------------------------------------------------------------------------------- /beyond/inpainting/scripts/test_rcblayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/scripts/test_rcblayer.py -------------------------------------------------------------------------------- /beyond/inpainting/src/INN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/INN.py -------------------------------------------------------------------------------- /beyond/inpainting/src/ImagineGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/ImagineGAN.py -------------------------------------------------------------------------------- /beyond/inpainting/src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/config.py -------------------------------------------------------------------------------- /beyond/inpainting/src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/dataset.py -------------------------------------------------------------------------------- /beyond/inpainting/src/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/masks.py -------------------------------------------------------------------------------- /beyond/inpainting/src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/metrics.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/atten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/atten.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/loss.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/model.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/networks.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/partialconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/partialconv.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/rcb_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/rcb_layer.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/reshape.py -------------------------------------------------------------------------------- /beyond/inpainting/src/model/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/model/transformer.py -------------------------------------------------------------------------------- /beyond/inpainting/src/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/tools.py -------------------------------------------------------------------------------- /beyond/inpainting/src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/utils.py -------------------------------------------------------------------------------- /beyond/inpainting/src/utilsgzh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/src/utilsgzh.py -------------------------------------------------------------------------------- /beyond/inpainting/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/test.py -------------------------------------------------------------------------------- /beyond/inpainting/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/inpainting/train.py -------------------------------------------------------------------------------- /beyond/relighting/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/.gitignore -------------------------------------------------------------------------------- /beyond/relighting/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/README.md -------------------------------------------------------------------------------- /beyond/relighting/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/data.py -------------------------------------------------------------------------------- /beyond/relighting/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/data/__init__.py -------------------------------------------------------------------------------- /beyond/relighting/data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/data/base_dataset.py -------------------------------------------------------------------------------- /beyond/relighting/data/dpr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/data/dpr_dataset.py -------------------------------------------------------------------------------- /beyond/relighting/data/dprtransfer_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/data/dprtransfer_dataset.py -------------------------------------------------------------------------------- /beyond/relighting/data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/data/image_folder.py -------------------------------------------------------------------------------- /beyond/relighting/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/data/loader.py -------------------------------------------------------------------------------- /beyond/relighting/datasets/randomdpr_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/datasets/randomdpr_test.txt -------------------------------------------------------------------------------- /beyond/relighting/datasets/randomdprtgt_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/datasets/randomdprtgt_test.txt -------------------------------------------------------------------------------- /beyond/relighting/datasets/train.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/datasets/train.lst -------------------------------------------------------------------------------- /beyond/relighting/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/models/__init__.py -------------------------------------------------------------------------------- /beyond/relighting/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/models/base_model.py -------------------------------------------------------------------------------- /beyond/relighting/models/base_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/models/base_networks.py -------------------------------------------------------------------------------- /beyond/relighting/models/dht_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/models/dht_model.py -------------------------------------------------------------------------------- /beyond/relighting/models/harmony_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/models/harmony_networks.py -------------------------------------------------------------------------------- /beyond/relighting/models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/models/position_encoding.py -------------------------------------------------------------------------------- /beyond/relighting/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/models/transformer.py -------------------------------------------------------------------------------- /beyond/relighting/options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/options/__init__.py -------------------------------------------------------------------------------- /beyond/relighting/options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/options/base_options.py -------------------------------------------------------------------------------- /beyond/relighting/options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/options/test_options.py -------------------------------------------------------------------------------- /beyond/relighting/options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/options/train_options.py -------------------------------------------------------------------------------- /beyond/relighting/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/test.py -------------------------------------------------------------------------------- /beyond/relighting/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/train.py -------------------------------------------------------------------------------- /beyond/relighting/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/train_net.py -------------------------------------------------------------------------------- /beyond/relighting/util/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/distributed.py -------------------------------------------------------------------------------- /beyond/relighting/util/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/evaluation.py -------------------------------------------------------------------------------- /beyond/relighting/util/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/flops.py -------------------------------------------------------------------------------- /beyond/relighting/util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/html.py -------------------------------------------------------------------------------- /beyond/relighting/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/loss.py -------------------------------------------------------------------------------- /beyond/relighting/util/misc copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/misc copy.py -------------------------------------------------------------------------------- /beyond/relighting/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/misc.py -------------------------------------------------------------------------------- /beyond/relighting/util/multiprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/multiprocessing.py -------------------------------------------------------------------------------- /beyond/relighting/util/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/ssim.py -------------------------------------------------------------------------------- /beyond/relighting/util/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/tools.py -------------------------------------------------------------------------------- /beyond/relighting/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/util.py -------------------------------------------------------------------------------- /beyond/relighting/util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/relighting/util/visualizer.py -------------------------------------------------------------------------------- /beyond/white_balance/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/README.md -------------------------------------------------------------------------------- /beyond/white_balance/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/__init__.py -------------------------------------------------------------------------------- /beyond/white_balance/data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/base_dataset.py -------------------------------------------------------------------------------- /beyond/white_balance/data/folds/fold1.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/folds/fold1.mat -------------------------------------------------------------------------------- /beyond/white_balance/data/folds/fold1_.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/folds/fold1_.mat -------------------------------------------------------------------------------- /beyond/white_balance/data/folds/fold2.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/folds/fold2.mat -------------------------------------------------------------------------------- /beyond/white_balance/data/folds/fold2_.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/folds/fold2_.mat -------------------------------------------------------------------------------- /beyond/white_balance/data/folds/fold3.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/folds/fold3.mat -------------------------------------------------------------------------------- /beyond/white_balance/data/folds/fold3_.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/folds/fold3_.mat -------------------------------------------------------------------------------- /beyond/white_balance/data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/image_folder.py -------------------------------------------------------------------------------- /beyond/white_balance/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/loader.py -------------------------------------------------------------------------------- /beyond/white_balance/data/set1test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/set1test_dataset.py -------------------------------------------------------------------------------- /beyond/white_balance/data/set_1_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/set_1_test.txt -------------------------------------------------------------------------------- /beyond/white_balance/data/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/train.txt -------------------------------------------------------------------------------- /beyond/white_balance/data/wb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/data/wb_dataset.py -------------------------------------------------------------------------------- /beyond/white_balance/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/models/__init__.py -------------------------------------------------------------------------------- /beyond/white_balance/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/models/base_model.py -------------------------------------------------------------------------------- /beyond/white_balance/models/base_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/models/base_networks.py -------------------------------------------------------------------------------- /beyond/white_balance/models/harmony_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/models/harmony_networks.py -------------------------------------------------------------------------------- /beyond/white_balance/models/ht_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/models/ht_model.py -------------------------------------------------------------------------------- /beyond/white_balance/models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/models/position_encoding.py -------------------------------------------------------------------------------- /beyond/white_balance/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/models/transformer.py -------------------------------------------------------------------------------- /beyond/white_balance/options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/options/__init__.py -------------------------------------------------------------------------------- /beyond/white_balance/options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/options/base_options.py -------------------------------------------------------------------------------- /beyond/white_balance/options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/options/test_options.py -------------------------------------------------------------------------------- /beyond/white_balance/options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/options/train_options.py -------------------------------------------------------------------------------- /beyond/white_balance/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/test.py -------------------------------------------------------------------------------- /beyond/white_balance/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/train.py -------------------------------------------------------------------------------- /beyond/white_balance/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/train_net.py -------------------------------------------------------------------------------- /beyond/white_balance/util/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/distributed.py -------------------------------------------------------------------------------- /beyond/white_balance/util/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/evaluation.py -------------------------------------------------------------------------------- /beyond/white_balance/util/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/flops.py -------------------------------------------------------------------------------- /beyond/white_balance/util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/html.py -------------------------------------------------------------------------------- /beyond/white_balance/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/loss.py -------------------------------------------------------------------------------- /beyond/white_balance/util/misc copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/misc copy.py -------------------------------------------------------------------------------- /beyond/white_balance/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/misc.py -------------------------------------------------------------------------------- /beyond/white_balance/util/multiprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/multiprocessing.py -------------------------------------------------------------------------------- /beyond/white_balance/util/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/ssim.py -------------------------------------------------------------------------------- /beyond/white_balance/util/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/tools.py -------------------------------------------------------------------------------- /beyond/white_balance/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/util.py -------------------------------------------------------------------------------- /beyond/white_balance/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/utils.py -------------------------------------------------------------------------------- /beyond/white_balance/util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/beyond/white_balance/util/visualizer.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/data/base_dataset.py -------------------------------------------------------------------------------- /data/ihd_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/data/ihd_dataset.py -------------------------------------------------------------------------------- /data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/data/image_folder.py -------------------------------------------------------------------------------- /data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/data/loader.py -------------------------------------------------------------------------------- /data/real_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/data/real_dataset.py -------------------------------------------------------------------------------- /evaluation/ih_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/evaluation/ih_evaluation.py -------------------------------------------------------------------------------- /evaluation/pytorch_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/evaluation/pytorch_ssim.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/base_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/base_networks.py -------------------------------------------------------------------------------- /models/cnnht_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/cnnht_model.py -------------------------------------------------------------------------------- /models/dht_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/dht_model.py -------------------------------------------------------------------------------- /models/harmony_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/harmony_networks.py -------------------------------------------------------------------------------- /models/ht_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/ht_model.py -------------------------------------------------------------------------------- /models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/position_encoding.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/models/transformer.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/options/__init__.py -------------------------------------------------------------------------------- /options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/options/base_options.py -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/options/train_options.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/train.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/train_net.py -------------------------------------------------------------------------------- /util/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/distributed.py -------------------------------------------------------------------------------- /util/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/evaluation.py -------------------------------------------------------------------------------- /util/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/flops.py -------------------------------------------------------------------------------- /util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/html.py -------------------------------------------------------------------------------- /util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/loss.py -------------------------------------------------------------------------------- /util/misc copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/misc copy.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/multiprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/multiprocessing.py -------------------------------------------------------------------------------- /util/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/ssim.py -------------------------------------------------------------------------------- /util/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/tools.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/util.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenglab/HarmonyTransformer/HEAD/util/visualizer.py --------------------------------------------------------------------------------