AdaInt
[CVPR 2022] Official PyTorch Implementation of "AdaInt: Learning Adaptive Intervals for 3D Lookup Tables on Real-time Image Enhancement" (https://arxiv.org/abs/2204.13983)
File Explorer
Download Latest Version (.zip)- build.yml
- publish-to-pypi.yml
- ailut_transform.cpp
- ailut_transform_cpu.cpp
- ailut_transform_cuda.cu
- __init__.py
- pyinterfaces.py
- version.py
- ailut-1.5.0-cp37-cp37m-linux_x86_64.whl
- setup.py
- test.txt
- train.txt
- train_input.txt
- train_label.txt
- test.txt
- train.txt
- train_aug.txt
- fivekrgb.py
- fivekxyz.py
- ppr10k.py
- fivek_calculate_metrics.m
- ppr10k_calculate_metrics.m
- __init__.py
- dataset.py
- demo.py
- model.py
- transforms.py
- Dockerfile
- __init__.py
- generation_inference.py
- inpainting_inference.py
- matting_inference.py
- restoration_face_inference.py
- restoration_inference.py
- restoration_video_inference.py
- test.py
- train.py
- __init__.py
- eval_hooks.py
- metric_utils.py
- metrics.py
- niqe_pris_params.npz
- __init__.py
- wrappers.py
- __init__.py
- ema.py
- visualization.py
- __init__.py
- builder.py
- __init__.py
- lr_updater.py
- __init__.py
- distributed_wrapper.py
- mask.py
- misc.py
- __init__.py
- augmentation.py
- blur_kernels.py
- compose.py
- crop.py
- formating.py
- generate_assistant.py
- loading.py
- matlab_like_resize.py
- matting_aug.py
- normalization.py
- random_degradations.py
- random_down_sampling.py
- utils.py
- __init__.py
- distributed_sampler.py
- __init__.py
- base_dataset.py
- base_generation_dataset.py
- base_matting_dataset.py
- base_sr_dataset.py
- builder.py
- comp1k_dataset.py
- dataset_wrappers.py
- generation_paired_dataset.py
- generation_unpaired_dataset.py
- img_inpainting_dataset.py
- registry.py
- sr_annotation_dataset.py
- sr_facial_landmark_dataset.py
- sr_folder_dataset.py
- sr_folder_gt_dataset.py
- sr_folder_multiple_gt_dataset.py
- sr_folder_ref_dataset.py
- sr_folder_video_dataset.py
- sr_lmdb_dataset.py
- sr_reds_dataset.py
- sr_reds_multiple_gt_dataset.py
- sr_test_multiple_gt_dataset.py
- sr_vid4_dataset.py
- sr_vimeo90k_dataset.py
- sr_vimeo90k_multiple_gt_dataset.py
- __init__.py
- deepfill_decoder.py
- fba_decoder.py
- gl_decoder.py
- indexnet_decoder.py
- pconv_decoder.py
- plain_decoder.py
- resnet_dec.py
- __init__.py
- deepfill_encoder.py
- fba_encoder.py
- gl_encoder.py
- indexnet_encoder.py
- pconv_encoder.py
- resnet.py
- resnet_enc.py
- vgg.py
- __init__.py
- contextual_attention_neck.py
- gl_dilation.py
- __init__.py
- gl_encoder_decoder.py
- pconv_encoder_decoder.py
- simple_encoder_decoder.py
- two_stage_encoder_decoder.py
- __init__.py
- resnet_generator.py
- unet_generator.py
- __init__.py
- basicvsr_net.py
- basicvsr_pp.py
- dic_net.py
- duf.py
- edsr.py
- edvr_net.py
- glean_styleganv2.py
- iconvsr.py
- liif_net.py
- rdn.py
- rrdb_net.py
- sr_resnet.py
- srcnn.py
- tdan_net.py
- tof.py
- ttsr_net.py
- __init__.py
- __init__.py
- aspp.py
- contextual_attention.py
- conv.py
- flow_warp.py
- gated_conv_module.py
- gca_module.py
- generation_model_utils.py
- img_normalize.py
- linear_module.py
- mask_conv_module.py
- model_utils.py
- partial_conv.py
- separable_conv_module.py
- sr_backbone_utils.py
- upsample.py
- __init__.py
- deepfill_disc.py
- gl_disc.py
- light_cnn.py
- modified_vgg.py
- multi_layer_disc.py
- patch_disc.py
- ttsr_disc.py
- __init__.py
- deepfill_refiner.py
- mlp_refiner.py
- plain_refiner.py
- __init__.py
- common.py
- generator_discriminator.py
- modules.py
- __init__.py
- __init__.py
- feedback_hour_glass.py
- lte.py
- __init__.py
- deepfillv1.py
- gl_inpaintor.py
- one_stage.py
- pconv_inpaintor.py
- two_stage.py
- __init__.py
- composition_loss.py
- feature_loss.py
- gan_loss.py
- gradient_loss.py
- perceptual_loss.py
- pixelwise_loss.py
- utils.py
- __init__.py
- base_mattor.py
- dim.py
- gca.py
- indexnet.py
- utils.py
- __init__.py
- basic_restorer.py
- basicvsr.py
- dic.py
- edvr.py
- esrgan.py
- glean.py
- liif.py
- real_esrgan.py
- srgan.py
- tdan.py
- ttsr.py
- __init__.py
- cycle_gan.py
- pix2pix.py
- __init__.py
- search_transformer.py
- __init__.py
- base.py
- builder.py
- registry.py
- __init__.py
- collect_env.py
- logger.py
- __init__.py
- version.py
- AiLUT-FiveK-sRGB.pth
- AiLUT-FiveK-XYZ.pth
- AiLUT-PPR10KA-sRGB.pth
- runtime.txt
- tests.txt
- a4739.jpg
- framework.jpg
- GT05.jpg
- GT26r.jpg
- 000001.png
- GT05.jpg
- baboon.png
- bbox_mask.png
- celeba_test.png
- test.png
- gl_test.py
- one_stage_gl.py
- pconv_inpaintor_test.py
- baboon_x4.png
- data.mdb
- lock.mdb
- meta_info.txt
- test.png
- test_single_ch.png
- GT05.jpg
- 3.jpg
- 1.jpg
- 2.jpg
- 00000000.png
- 00000001.png
- 00000000.png
- GT05.png
- 5.jpg
- 6.jpg
- 1.jpg
- 2.jpg
- 3.jpg
- 4.jpg
- im1.png
- im2.png
- im3.png
- im4.png
- im5.png
- im6.png
- im7.png
- facemark_ann.npy
- mask_list.txt
- mask_list_single_ch.txt
- test_list.json
- train.txt
- test_generation_datasets.py
- test_matting_datasets.py
- test_repeat_dataset.py
- test_sr_dataset.py
- test_augmentation.py
- test_compose.py
- test_crop.py
- test_formating.py
- test_generate_assistant.py
- test_loading.py
- test_matlab_resize.py
- test_normalization.py
- test_pipeline_utils.py
- test_random_degradations.py
- test_random_down_sampling.py
- test_trimap.py
- test_metrics.py
- test_decoders.py
- test_deepfill_decoder.py
- test_deepfill_encdec.py
- test_deepfill_encoder.py
- test_encoder_decoder.py
- test_encoders.py
- test_gl_model.py
- test_pconv_encdec.py
- test_generators.py
- test_basicvsr_net.py
- test_basicvsr_plusplus.py
- test_dic_net.py
- test_duf.py
- test_edvr_net.py
- test_glean_net.py
- test_iconvsr.py
- test_liif_net.py
- test_rdn.py
- test_sr_backbones.py
- test_tdan_net.py
- test_tof.py
- test_common_module.py
- test_flow_warp.py
- test_img_normalize.py
- test_model_utils.py
- test_deepfill_disc.py
- test_discriminators.py
- test_light_cnn.py
- test_multi_layer_disc.py
- test_deepfill_refiner.py
- test_matting_refiners.py
- test_mlp_refiner.py
- test_stylegan2.py
- test_feedback_hour_glass.py
- test_lte.py
- test_deepfill_inpaintor.py
- test_gl_inpaintor.py
- test_one_stage_inpaintor.py
- test_pconv_inpaintor.py
- test_two_stage_inpaintor.py
- test_feature_loss.py
- test_losses.py
- test_mattors.py
- test_basic_restorer.py
- test_basicvsr_model.py
- test_dic_model.py
- test_edvr_model.py
- test_esrgan.py
- test_glean.py
- test_liif.py
- test_real_esrgan.py
- test_srgan.py
- test_tdan.py
- test_ttsr.py
- test_cyclegan.py
- test_search_transformer.py
- test_base_model.py
- test_dataset_builder.py
- test_ema_hook.py
- test_eval_hook.py
- test_optimizer.py
- test_visual_hook.py
- test_mask_generation.py
- test_onnx_wraper.py
- test_pix2pix.py
- test_tensor2img.py
- test_inference.py
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- preprocess_bgm_dataset.py
- check_extended_fg.py
- extend_fg.py
- filter_comp1k_anno.py
- preprocess_comp1k_dataset.py
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- preprocess_df2k_ost_dataset.py
- README.md
- README_zh-CN.md
- preprocess_div2k_dataset.py
- README.md
- README_zh-CN.md
- preprocess_reds_dataset.py
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- preprocess_vimeo90k_dataset.py
- README.md
- README_zh-CN.md
- README.md
- README_zh-CN.md
- mmedit2torchserve.py
- mmedit_handler.py
- test_torchserver.py
- deploy_test.py
- dist_test.sh
- dist_train.sh
- evaluate_comp1k.py
- get_flops.py
- onnx2tensorrt.py
- publish_model.py
- pytorch2onnx.py
- slurm_test.sh
- slurm_train.sh
- test.py
- train.py
- .gitignore
- .pre-commit-config.yaml
- .pylintrc
- CITATION.cff
- LICENSE
- Prepare_FiveK.md
- README.md
- requirements.txt
- setup.cfg
- setup.py
# Installation Guide
1. Get the code
git clone https://github.com/ImCharlesY/AdaInt
Downloads the entire project code from GitHub to your computer.
cd AdaInt
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install -v -e .
Installs the package published on PyPI directly β no need to clone the source.
pip install adaint/ailut_transform/ailut-1.5.0-cp37-cp37m-linux_x86_64.whl
Installs the package published on PyPI directly β no need to clone the source.
After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.
Pulled directly from this repo's README.
3. Docker
EasyPrerequisites
- Git Needed to download the project code from GitHub.
- Docker Desktop Needed to build and run containers. Install it and keep it running in the background.
docker build -f docker/Dockerfile -t adaint .
Builds a runnable image based on the Dockerfile.
docker run -p 8080:80 adaint
Runs the built image as an actual container.
Run docker compose ps to check the containers are Up. If the README mentions a port, open http://localhost:PORT in your browser.
4. Python
EasyPrerequisites
pip install -r requirements.txt
Installs the Python libraries listed in requirements.txt (or similar).
pip install -v -e .
Installs the package published on PyPI directly β no need to clone the source.
pip install adaint/ailut_transform/ailut-1.5.0-cp37-cp37m-linux_x86_64.whl
Installs the package published on PyPI directly β no need to clone the source.
pip uninstall ailut
Type this command into your terminal and run it.
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)