├── .gitattributes ├── .github ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE │ ├── bug_template.md │ ├── config.yml │ └── enhancement_template.md ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── build-and-test.yml │ └── publish-package.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── NOTICE ├── README.md ├── WEIGHTS_LICENSE ├── hubconf.py ├── neuralcompression ├── __init__.py ├── _outputs │ ├── __init__.py │ ├── _discriminator_output.py │ ├── _hyperprior_output.py │ └── _vqvae_autoencoder_output.py ├── data │ ├── __init__.py │ ├── _clic_2020_image.py │ ├── _clic_2020_video.py │ ├── _div2k_dataset.py │ ├── _kodak.py │ ├── _openimages_v6.py │ └── _vimeo_90k_septuplet.py ├── distributions │ ├── __init__.py │ ├── _noisy_normal.py │ └── _uniform_noise.py ├── functional │ ├── __init__.py │ ├── _count_flops.py │ ├── _dense_image_warp.py │ ├── _estimate_tails.py │ ├── _hsv_to_rgb.py │ ├── _image_to_255_scale.py │ ├── _information_content.py │ ├── _log_cdf.py │ ├── _log_expm1.py │ ├── _log_ndtr.py │ ├── _log_survival_function.py │ ├── _lower_bound.py │ ├── _lower_tail.py │ ├── _lpips.py │ ├── _multiscale_structural_similarity.py │ ├── _ndtr.py │ ├── _optical_flow_to_color.py │ ├── _pad_image.py │ ├── _quantization_offset.py │ ├── _soft_round.py │ ├── _soft_round_conditional_mean.py │ ├── _soft_round_inverse.py │ ├── _survival_function.py │ └── _upper_tail.py ├── layers │ ├── __init__.py │ ├── _channel_norm.py │ ├── _continuous_entropy.py │ ├── _normalize_latent.py │ ├── _project_latent.py │ ├── _q_bottleneck.py │ ├── _simplified_gdn.py │ ├── _vq_bottleneck.py │ └── _xca.py ├── loss_fn │ ├── __init__.py │ ├── _binary_crossentropy_loss.py │ ├── _distortion_loss.py │ ├── _gan_losses.py │ ├── _mse_loss.py │ ├── _mse_lpips_loss.py │ ├── _normfix_lpips.py │ ├── _oasis_loss.py │ └── _target_rate_config.py ├── metrics │ ├── __init__.py │ ├── _calc_psnr.py │ ├── _dists.py │ ├── _fid.py │ ├── _fid_swav.py │ ├── _kid.py │ ├── _msssim.py │ ├── _pickle_size.py │ └── _update_patch_fid.py ├── models │ ├── __init__.py │ ├── _discriminator.py │ ├── _hific_autoencoder.py │ ├── _hific_encoder_decoder.py │ ├── _hyperprior_autoencoder.py │ ├── _unet_discriminator.py │ ├── _vqvae_projector.py │ ├── _vqvae_xcit_autoencoder.py │ └── deep_video_compression.py ├── optim │ ├── __init__.py │ ├── _identity.py │ ├── _ramp_const_cosine.py │ └── _ramp_cosine.py └── zoo │ ├── __init__.py │ ├── _msillm.py │ ├── _msillm_vqvae.py │ └── _noganms.py ├── projects ├── README.md ├── bits_back_diffusion │ ├── .gitignore │ ├── README.md │ ├── bits_back_diffusion │ │ ├── __init__.py │ │ ├── codec.py │ │ └── script_util.py │ ├── compute_vlb.py │ ├── config │ │ ├── compute_vlb.yaml │ │ ├── data │ │ │ ├── cifar.yaml │ │ │ └── imagenet.yaml │ │ ├── default.yaml │ │ ├── diffusion │ │ │ ├── default.yaml │ │ │ └── respaced.yaml │ │ ├── encode.yaml │ │ ├── evaluator │ │ │ ├── cifar.yaml │ │ │ └── imagenet.yaml │ │ ├── mode │ │ │ └── submitit.yaml │ │ ├── model │ │ │ ├── cifar.yaml │ │ │ └── imagenet.yaml │ │ └── train.yaml │ ├── datasets │ │ └── cifar10.py │ ├── encode.py │ ├── environment.yml │ ├── test.py │ └── train.py ├── deep_video_compression │ ├── README.md │ ├── _utils.py │ ├── config │ │ ├── base.yaml │ │ ├── mode │ │ │ ├── submitit_multi_node.yaml │ │ │ └── submitit_single_node.yaml │ │ └── training_stages │ │ │ ├── s1_motion_estimation.yaml │ │ │ ├── s2_motion_compression.yaml │ │ │ ├── s3_motion_compensation.yaml │ │ │ ├── s4_total_2frame.yaml │ │ │ └── s5_total.yaml │ ├── data_module.py │ ├── dvc_module.py │ ├── requirements.txt │ └── train.py ├── illm │ ├── README.md │ ├── conf │ │ ├── config.yaml │ │ ├── data │ │ │ └── local_data_openimages.yaml │ │ ├── discriminator │ │ │ └── condunet_ch1025_factor8_context220.yaml │ │ ├── distortion_loss │ │ │ ├── mse_distortion.yaml │ │ │ └── mse_lpips.yaml │ │ ├── image_logs │ │ │ └── default_logs.yaml │ │ ├── latent_projector │ │ │ └── vqvae_xcit_p8_ch64_cb1024_h8.yaml │ │ ├── mode │ │ │ └── submitit_multi_node.yaml │ │ ├── model │ │ │ └── hific_autoencoder.yaml │ │ ├── optimizer │ │ │ ├── model_adamw.yaml │ │ │ └── model_adamw_disc_const.yaml │ │ ├── pretrained_autoencoder │ │ │ ├── example.yaml │ │ │ └── nogan_target0.14bpp.yaml │ │ ├── pretrained_latent_autoencoder │ │ │ ├── 20221108_vqvae_xcit_p8_ch64_cb1024_h8.yaml │ │ │ └── example_local.yaml │ │ ├── rate_target │ │ │ ├── notarget.yaml │ │ │ ├── target-0.00875.yaml │ │ │ ├── target-0.0175.yaml │ │ │ ├── target-0.035.yaml │ │ │ ├── target-0.07.yaml │ │ │ ├── target-0.14.yaml │ │ │ ├── target-0.3.yaml │ │ │ ├── target-0.45.yaml │ │ │ └── target-0.9.yaml │ │ ├── test_dataloaders │ │ │ └── test.yaml │ │ └── trainer │ │ │ └── default_trainer.yaml │ ├── default_transforms.py │ ├── eval_folder_example.py │ ├── gan_compression_module.py │ ├── image_module.py │ ├── open_images_datamodule.py │ ├── requirements.txt │ ├── results.csv │ ├── target_compression_module.py │ └── train.py ├── jax_entropy_coders │ ├── README.md │ ├── craystack │ │ ├── __init__.py │ │ ├── _backend.py │ │ ├── codecs.py │ │ └── coder.py │ ├── jax_arithmetic_coder.py │ ├── requirements.txt │ └── test_entropy_coders.py └── torch_vct │ ├── .gitignore │ ├── README.md │ ├── __init__.py │ ├── config │ ├── analysis_transform │ │ ├── elic.yaml │ │ └── gdn.yaml │ ├── datamodule │ │ ├── kinetics.yaml │ │ └── vimeo.yaml │ ├── mode │ │ └── submitit_multi_node.yaml │ ├── optim │ │ ├── adam.yaml │ │ ├── adamw.yaml │ │ └── lion.yaml │ ├── scheduler │ │ ├── cosine.yaml │ │ ├── exponential.yaml │ │ ├── linearrampcosine.yaml │ │ ├── linearramplinear.yaml │ │ └── linearwarmupcosine.yaml │ ├── synthesis_transform │ │ ├── elic.yaml │ │ └── gdn.yaml │ ├── test_datamodule │ │ └── uvg.yaml │ └── train_config.yaml │ ├── datamodules │ ├── __init__.py │ ├── kinetics.py │ ├── uvg.py │ ├── video_data_api.py │ └── vimeo.py │ ├── environment.yml │ ├── fig │ └── vct_rd.png │ ├── model_lightning.py │ ├── model_pipeline.py │ ├── model_train.py │ ├── neural │ ├── __init__.py │ ├── bottlenecks.py │ ├── entropy_model.py │ ├── entropy_model_layers.py │ ├── layers_utils.py │ ├── patcher.py │ └── transforms.py │ ├── tests │ ├── __init__.py │ └── neural │ │ ├── __init__.py │ │ ├── test_entropy_model_layers.py │ │ ├── test_layers_utils.py │ │ └── test_patcher.py │ └── utils │ ├── __init__.py │ ├── checkpoint_loaders.py │ ├── datavis.py │ ├── hydra_tools.py │ ├── memoize.py │ └── schedulers.py ├── pyproject.toml ├── setup.cfg ├── setup.py ├── tests ├── cached_data │ ├── create_cached_tensorflow_data.py │ └── dense_image_warp.pkl ├── conftest.py ├── data │ ├── __init__.py │ ├── test_clic_2020_image.py │ ├── test_clic_2020_video.py │ ├── test_div2k.py │ ├── test_kodak.py │ ├── test_open_images.py │ └── test_vimeo_90k_septuplet.py ├── distributions │ ├── __init__.py │ ├── test_noisy_normal.py │ └── test_uniform_noise.py ├── functional │ ├── __init__.py │ ├── test_count_flops.py │ ├── test_dense_image_warp.py │ ├── test_estimate_tails.py │ ├── test_hsv_to_rgb.py │ ├── test_image_255_scale.py │ ├── test_information_content.py │ ├── test_log_cdf.py │ ├── test_log_expm1.py │ ├── test_log_ndtr.py │ ├── test_log_survival_function.py │ ├── test_lower_bound.py │ ├── test_lower_tail.py │ ├── test_ndtr.py │ ├── test_optical_flow_to_rgb.py │ ├── test_pad_image.py │ ├── test_quantization_offset.py │ ├── test_soft_round.py │ ├── test_soft_round_conditional_mean.py │ ├── test_soft_round_inverse.py │ ├── test_survival_function.py │ └── test_upper_tail.py ├── layers │ ├── __init__.py │ ├── test_channelnorm.py │ └── test_continuous_entropy.py ├── loss_fn │ ├── test_binary_cross_entropy.py │ ├── test_mse_loss.py │ ├── test_mse_lpips_loss.py │ └── test_oasis_loss.py ├── metrics │ ├── __init__.py │ ├── test_calc_psnr.py │ ├── test_dists.py │ ├── test_fid.py │ ├── test_kid.py │ ├── test_msssim.py │ ├── test_pickle_size.py │ ├── test_swav_fid.py │ └── test_update_patch_fid.py ├── models │ ├── __init__.py │ ├── test_hific_autoencoder.py │ ├── test_hific_encoder.py │ ├── test_hific_generator.py │ └── test_unet_discriminator.py ├── test_layers_gdn.py ├── test_models.py └── utils.py └── tutorials ├── Flop_Count_Example.ipynb ├── Metrics_Example.ipynb └── image_compression.ipynb /.gitattributes: -------------------------------------------------------------------------------- 1 | *.ipynb linguist-documentation 2 | -------------------------------------------------------------------------------- /.github/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/ISSUE_TEMPLATE/bug_template.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/enhancement_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/ISSUE_TEMPLATE/enhancement_template.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/workflows/build-and-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/workflows/build-and-test.yml -------------------------------------------------------------------------------- /.github/workflows/publish-package.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.github/workflows/publish-package.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/README.md -------------------------------------------------------------------------------- /WEIGHTS_LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/WEIGHTS_LICENSE -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/hubconf.py -------------------------------------------------------------------------------- /neuralcompression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/__init__.py -------------------------------------------------------------------------------- /neuralcompression/_outputs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/_outputs/__init__.py -------------------------------------------------------------------------------- /neuralcompression/_outputs/_discriminator_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/_outputs/_discriminator_output.py -------------------------------------------------------------------------------- /neuralcompression/_outputs/_hyperprior_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/_outputs/_hyperprior_output.py -------------------------------------------------------------------------------- /neuralcompression/_outputs/_vqvae_autoencoder_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/_outputs/_vqvae_autoencoder_output.py -------------------------------------------------------------------------------- /neuralcompression/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/data/__init__.py -------------------------------------------------------------------------------- /neuralcompression/data/_clic_2020_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/data/_clic_2020_image.py -------------------------------------------------------------------------------- /neuralcompression/data/_clic_2020_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/data/_clic_2020_video.py -------------------------------------------------------------------------------- /neuralcompression/data/_div2k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/data/_div2k_dataset.py -------------------------------------------------------------------------------- /neuralcompression/data/_kodak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/data/_kodak.py -------------------------------------------------------------------------------- /neuralcompression/data/_openimages_v6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/data/_openimages_v6.py -------------------------------------------------------------------------------- /neuralcompression/data/_vimeo_90k_septuplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/data/_vimeo_90k_septuplet.py -------------------------------------------------------------------------------- /neuralcompression/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/distributions/__init__.py -------------------------------------------------------------------------------- /neuralcompression/distributions/_noisy_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/distributions/_noisy_normal.py -------------------------------------------------------------------------------- /neuralcompression/distributions/_uniform_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/distributions/_uniform_noise.py -------------------------------------------------------------------------------- /neuralcompression/functional/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/__init__.py -------------------------------------------------------------------------------- /neuralcompression/functional/_count_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_count_flops.py -------------------------------------------------------------------------------- /neuralcompression/functional/_dense_image_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_dense_image_warp.py -------------------------------------------------------------------------------- /neuralcompression/functional/_estimate_tails.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_estimate_tails.py -------------------------------------------------------------------------------- /neuralcompression/functional/_hsv_to_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_hsv_to_rgb.py -------------------------------------------------------------------------------- /neuralcompression/functional/_image_to_255_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_image_to_255_scale.py -------------------------------------------------------------------------------- /neuralcompression/functional/_information_content.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_information_content.py -------------------------------------------------------------------------------- /neuralcompression/functional/_log_cdf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_log_cdf.py -------------------------------------------------------------------------------- /neuralcompression/functional/_log_expm1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_log_expm1.py -------------------------------------------------------------------------------- /neuralcompression/functional/_log_ndtr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_log_ndtr.py -------------------------------------------------------------------------------- /neuralcompression/functional/_log_survival_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_log_survival_function.py -------------------------------------------------------------------------------- /neuralcompression/functional/_lower_bound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_lower_bound.py -------------------------------------------------------------------------------- /neuralcompression/functional/_lower_tail.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_lower_tail.py -------------------------------------------------------------------------------- /neuralcompression/functional/_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_lpips.py -------------------------------------------------------------------------------- /neuralcompression/functional/_multiscale_structural_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_multiscale_structural_similarity.py -------------------------------------------------------------------------------- /neuralcompression/functional/_ndtr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_ndtr.py -------------------------------------------------------------------------------- /neuralcompression/functional/_optical_flow_to_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_optical_flow_to_color.py -------------------------------------------------------------------------------- /neuralcompression/functional/_pad_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_pad_image.py -------------------------------------------------------------------------------- /neuralcompression/functional/_quantization_offset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_quantization_offset.py -------------------------------------------------------------------------------- /neuralcompression/functional/_soft_round.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_soft_round.py -------------------------------------------------------------------------------- /neuralcompression/functional/_soft_round_conditional_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_soft_round_conditional_mean.py -------------------------------------------------------------------------------- /neuralcompression/functional/_soft_round_inverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_soft_round_inverse.py -------------------------------------------------------------------------------- /neuralcompression/functional/_survival_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_survival_function.py -------------------------------------------------------------------------------- /neuralcompression/functional/_upper_tail.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/functional/_upper_tail.py -------------------------------------------------------------------------------- /neuralcompression/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/__init__.py -------------------------------------------------------------------------------- /neuralcompression/layers/_channel_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_channel_norm.py -------------------------------------------------------------------------------- /neuralcompression/layers/_continuous_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_continuous_entropy.py -------------------------------------------------------------------------------- /neuralcompression/layers/_normalize_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_normalize_latent.py -------------------------------------------------------------------------------- /neuralcompression/layers/_project_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_project_latent.py -------------------------------------------------------------------------------- /neuralcompression/layers/_q_bottleneck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_q_bottleneck.py -------------------------------------------------------------------------------- /neuralcompression/layers/_simplified_gdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_simplified_gdn.py -------------------------------------------------------------------------------- /neuralcompression/layers/_vq_bottleneck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_vq_bottleneck.py -------------------------------------------------------------------------------- /neuralcompression/layers/_xca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/layers/_xca.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/__init__.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_binary_crossentropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_binary_crossentropy_loss.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_distortion_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_distortion_loss.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_gan_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_gan_losses.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_mse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_mse_loss.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_mse_lpips_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_mse_lpips_loss.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_normfix_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_normfix_lpips.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_oasis_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_oasis_loss.py -------------------------------------------------------------------------------- /neuralcompression/loss_fn/_target_rate_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/loss_fn/_target_rate_config.py -------------------------------------------------------------------------------- /neuralcompression/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/__init__.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_calc_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_calc_psnr.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_dists.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_fid.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_fid_swav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_fid_swav.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_kid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_kid.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_msssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_msssim.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_pickle_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_pickle_size.py -------------------------------------------------------------------------------- /neuralcompression/metrics/_update_patch_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/metrics/_update_patch_fid.py -------------------------------------------------------------------------------- /neuralcompression/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/__init__.py -------------------------------------------------------------------------------- /neuralcompression/models/_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/_discriminator.py -------------------------------------------------------------------------------- /neuralcompression/models/_hific_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/_hific_autoencoder.py -------------------------------------------------------------------------------- /neuralcompression/models/_hific_encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/_hific_encoder_decoder.py -------------------------------------------------------------------------------- /neuralcompression/models/_hyperprior_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/_hyperprior_autoencoder.py -------------------------------------------------------------------------------- /neuralcompression/models/_unet_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/_unet_discriminator.py -------------------------------------------------------------------------------- /neuralcompression/models/_vqvae_projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/_vqvae_projector.py -------------------------------------------------------------------------------- /neuralcompression/models/_vqvae_xcit_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/_vqvae_xcit_autoencoder.py -------------------------------------------------------------------------------- /neuralcompression/models/deep_video_compression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/models/deep_video_compression.py -------------------------------------------------------------------------------- /neuralcompression/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/optim/__init__.py -------------------------------------------------------------------------------- /neuralcompression/optim/_identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/optim/_identity.py -------------------------------------------------------------------------------- /neuralcompression/optim/_ramp_const_cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/optim/_ramp_const_cosine.py -------------------------------------------------------------------------------- /neuralcompression/optim/_ramp_cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/optim/_ramp_cosine.py -------------------------------------------------------------------------------- /neuralcompression/zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/zoo/__init__.py -------------------------------------------------------------------------------- /neuralcompression/zoo/_msillm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/zoo/_msillm.py -------------------------------------------------------------------------------- /neuralcompression/zoo/_msillm_vqvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/zoo/_msillm_vqvae.py -------------------------------------------------------------------------------- /neuralcompression/zoo/_noganms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/neuralcompression/zoo/_noganms.py -------------------------------------------------------------------------------- /projects/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/README.md -------------------------------------------------------------------------------- /projects/bits_back_diffusion/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/.gitignore -------------------------------------------------------------------------------- /projects/bits_back_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/README.md -------------------------------------------------------------------------------- /projects/bits_back_diffusion/bits_back_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/bits_back_diffusion/bits_back_diffusion/codec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/bits_back_diffusion/codec.py -------------------------------------------------------------------------------- /projects/bits_back_diffusion/bits_back_diffusion/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/bits_back_diffusion/script_util.py -------------------------------------------------------------------------------- /projects/bits_back_diffusion/compute_vlb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/compute_vlb.py -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/compute_vlb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/compute_vlb.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/data/cifar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/data/cifar.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/data/imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/data/imagenet.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/default.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/diffusion/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/diffusion/default.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/diffusion/respaced.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/diffusion/respaced.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/encode.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/encode.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/evaluator/cifar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/evaluator/cifar.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/evaluator/imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/evaluator/imagenet.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/mode/submitit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/mode/submitit.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/model/cifar.yaml: -------------------------------------------------------------------------------- 1 | dropout: 0.3 2 | num_res_blocks: 3 3 | 4 | defaults: 5 | - imagenet 6 | - _self_ 7 | -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/model/imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/model/imagenet.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/config/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/config/train.yaml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/datasets/cifar10.py -------------------------------------------------------------------------------- /projects/bits_back_diffusion/encode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/encode.py -------------------------------------------------------------------------------- /projects/bits_back_diffusion/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/environment.yml -------------------------------------------------------------------------------- /projects/bits_back_diffusion/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/test.py -------------------------------------------------------------------------------- /projects/bits_back_diffusion/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/bits_back_diffusion/train.py -------------------------------------------------------------------------------- /projects/deep_video_compression/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/README.md -------------------------------------------------------------------------------- /projects/deep_video_compression/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/_utils.py -------------------------------------------------------------------------------- /projects/deep_video_compression/config/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/base.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/config/mode/submitit_multi_node.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/mode/submitit_multi_node.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/config/mode/submitit_single_node.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/mode/submitit_single_node.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/config/training_stages/s1_motion_estimation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/training_stages/s1_motion_estimation.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/config/training_stages/s2_motion_compression.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/training_stages/s2_motion_compression.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/config/training_stages/s3_motion_compensation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/training_stages/s3_motion_compensation.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/config/training_stages/s4_total_2frame.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/training_stages/s4_total_2frame.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/config/training_stages/s5_total.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/config/training_stages/s5_total.yaml -------------------------------------------------------------------------------- /projects/deep_video_compression/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/data_module.py -------------------------------------------------------------------------------- /projects/deep_video_compression/dvc_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/dvc_module.py -------------------------------------------------------------------------------- /projects/deep_video_compression/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/requirements.txt -------------------------------------------------------------------------------- /projects/deep_video_compression/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/deep_video_compression/train.py -------------------------------------------------------------------------------- /projects/illm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/README.md -------------------------------------------------------------------------------- /projects/illm/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/config.yaml -------------------------------------------------------------------------------- /projects/illm/conf/data/local_data_openimages.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/data/local_data_openimages.yaml -------------------------------------------------------------------------------- /projects/illm/conf/discriminator/condunet_ch1025_factor8_context220.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/discriminator/condunet_ch1025_factor8_context220.yaml -------------------------------------------------------------------------------- /projects/illm/conf/distortion_loss/mse_distortion.yaml: -------------------------------------------------------------------------------- 1 | _target_: neuralcompression.loss_fn.MSELoss 2 | scale_param: 65025.0 # 255 ** 2 3 | -------------------------------------------------------------------------------- /projects/illm/conf/distortion_loss/mse_lpips.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/distortion_loss/mse_lpips.yaml -------------------------------------------------------------------------------- /projects/illm/conf/image_logs/default_logs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/image_logs/default_logs.yaml -------------------------------------------------------------------------------- /projects/illm/conf/latent_projector/vqvae_xcit_p8_ch64_cb1024_h8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/latent_projector/vqvae_xcit_p8_ch64_cb1024_h8.yaml -------------------------------------------------------------------------------- /projects/illm/conf/mode/submitit_multi_node.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/mode/submitit_multi_node.yaml -------------------------------------------------------------------------------- /projects/illm/conf/model/hific_autoencoder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/model/hific_autoencoder.yaml -------------------------------------------------------------------------------- /projects/illm/conf/optimizer/model_adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/optimizer/model_adamw.yaml -------------------------------------------------------------------------------- /projects/illm/conf/optimizer/model_adamw_disc_const.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/optimizer/model_adamw_disc_const.yaml -------------------------------------------------------------------------------- /projects/illm/conf/pretrained_autoencoder/example.yaml: -------------------------------------------------------------------------------- 1 | path: null 2 | -------------------------------------------------------------------------------- /projects/illm/conf/pretrained_autoencoder/nogan_target0.14bpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/pretrained_autoencoder/nogan_target0.14bpp.yaml -------------------------------------------------------------------------------- /projects/illm/conf/pretrained_latent_autoencoder/20221108_vqvae_xcit_p8_ch64_cb1024_h8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/pretrained_latent_autoencoder/20221108_vqvae_xcit_p8_ch64_cb1024_h8.yaml -------------------------------------------------------------------------------- /projects/illm/conf/pretrained_latent_autoencoder/example_local.yaml: -------------------------------------------------------------------------------- 1 | path: null 2 | -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/notarget.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/notarget.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.00875.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.00875.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.0175.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.0175.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.035.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.035.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.07.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.07.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.14.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.3.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.45.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.45.yaml -------------------------------------------------------------------------------- /projects/illm/conf/rate_target/target-0.9.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/rate_target/target-0.9.yaml -------------------------------------------------------------------------------- /projects/illm/conf/test_dataloaders/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/test_dataloaders/test.yaml -------------------------------------------------------------------------------- /projects/illm/conf/trainer/default_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/conf/trainer/default_trainer.yaml -------------------------------------------------------------------------------- /projects/illm/default_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/default_transforms.py -------------------------------------------------------------------------------- /projects/illm/eval_folder_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/eval_folder_example.py -------------------------------------------------------------------------------- /projects/illm/gan_compression_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/gan_compression_module.py -------------------------------------------------------------------------------- /projects/illm/image_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/image_module.py -------------------------------------------------------------------------------- /projects/illm/open_images_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/open_images_datamodule.py -------------------------------------------------------------------------------- /projects/illm/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/requirements.txt -------------------------------------------------------------------------------- /projects/illm/results.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/results.csv -------------------------------------------------------------------------------- /projects/illm/target_compression_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/target_compression_module.py -------------------------------------------------------------------------------- /projects/illm/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/illm/train.py -------------------------------------------------------------------------------- /projects/jax_entropy_coders/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/README.md -------------------------------------------------------------------------------- /projects/jax_entropy_coders/craystack/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/craystack/__init__.py -------------------------------------------------------------------------------- /projects/jax_entropy_coders/craystack/_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/craystack/_backend.py -------------------------------------------------------------------------------- /projects/jax_entropy_coders/craystack/codecs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/craystack/codecs.py -------------------------------------------------------------------------------- /projects/jax_entropy_coders/craystack/coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/craystack/coder.py -------------------------------------------------------------------------------- /projects/jax_entropy_coders/jax_arithmetic_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/jax_arithmetic_coder.py -------------------------------------------------------------------------------- /projects/jax_entropy_coders/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/requirements.txt -------------------------------------------------------------------------------- /projects/jax_entropy_coders/test_entropy_coders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/jax_entropy_coders/test_entropy_coders.py -------------------------------------------------------------------------------- /projects/torch_vct/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | .vscode 3 | lightning_logs 4 | *.ipynb 5 | -------------------------------------------------------------------------------- /projects/torch_vct/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/README.md -------------------------------------------------------------------------------- /projects/torch_vct/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/torch_vct/config/analysis_transform/elic.yaml: -------------------------------------------------------------------------------- 1 | _target_: neural.transforms.ELICAnalysis 2 | compression_channels: 192 3 | -------------------------------------------------------------------------------- /projects/torch_vct/config/analysis_transform/gdn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/analysis_transform/gdn.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/datamodule/kinetics.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/datamodule/kinetics.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/datamodule/vimeo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/datamodule/vimeo.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/mode/submitit_multi_node.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/mode/submitit_multi_node.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/optim/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/optim/adam.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/optim/adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/optim/adamw.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/optim/lion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/optim/lion.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/scheduler/cosine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/scheduler/cosine.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/scheduler/exponential.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.optim.lr_scheduler.ExponentialLR 2 | gamma: 0.92 3 | -------------------------------------------------------------------------------- /projects/torch_vct/config/scheduler/linearrampcosine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/scheduler/linearrampcosine.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/scheduler/linearramplinear.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/scheduler/linearramplinear.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/scheduler/linearwarmupcosine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/scheduler/linearwarmupcosine.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/synthesis_transform/elic.yaml: -------------------------------------------------------------------------------- 1 | _target_: neural.transforms.ELICSynthesis 2 | output_channels: 3 3 | -------------------------------------------------------------------------------- /projects/torch_vct/config/synthesis_transform/gdn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/synthesis_transform/gdn.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/test_datamodule/uvg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/test_datamodule/uvg.yaml -------------------------------------------------------------------------------- /projects/torch_vct/config/train_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/config/train_config.yaml -------------------------------------------------------------------------------- /projects/torch_vct/datamodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/torch_vct/datamodules/kinetics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/datamodules/kinetics.py -------------------------------------------------------------------------------- /projects/torch_vct/datamodules/uvg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/datamodules/uvg.py -------------------------------------------------------------------------------- /projects/torch_vct/datamodules/video_data_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/datamodules/video_data_api.py -------------------------------------------------------------------------------- /projects/torch_vct/datamodules/vimeo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/datamodules/vimeo.py -------------------------------------------------------------------------------- /projects/torch_vct/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/environment.yml -------------------------------------------------------------------------------- /projects/torch_vct/fig/vct_rd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/fig/vct_rd.png -------------------------------------------------------------------------------- /projects/torch_vct/model_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/model_lightning.py -------------------------------------------------------------------------------- /projects/torch_vct/model_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/model_pipeline.py -------------------------------------------------------------------------------- /projects/torch_vct/model_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/model_train.py -------------------------------------------------------------------------------- /projects/torch_vct/neural/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/torch_vct/neural/bottlenecks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/neural/bottlenecks.py -------------------------------------------------------------------------------- /projects/torch_vct/neural/entropy_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/neural/entropy_model.py -------------------------------------------------------------------------------- /projects/torch_vct/neural/entropy_model_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/neural/entropy_model_layers.py -------------------------------------------------------------------------------- /projects/torch_vct/neural/layers_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/neural/layers_utils.py -------------------------------------------------------------------------------- /projects/torch_vct/neural/patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/neural/patcher.py -------------------------------------------------------------------------------- /projects/torch_vct/neural/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/neural/transforms.py -------------------------------------------------------------------------------- /projects/torch_vct/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/torch_vct/tests/neural/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/torch_vct/tests/neural/test_entropy_model_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/tests/neural/test_entropy_model_layers.py -------------------------------------------------------------------------------- /projects/torch_vct/tests/neural/test_layers_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/tests/neural/test_layers_utils.py -------------------------------------------------------------------------------- /projects/torch_vct/tests/neural/test_patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/tests/neural/test_patcher.py -------------------------------------------------------------------------------- /projects/torch_vct/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/torch_vct/utils/checkpoint_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/utils/checkpoint_loaders.py -------------------------------------------------------------------------------- /projects/torch_vct/utils/datavis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/utils/datavis.py -------------------------------------------------------------------------------- /projects/torch_vct/utils/hydra_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/utils/hydra_tools.py -------------------------------------------------------------------------------- /projects/torch_vct/utils/memoize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/utils/memoize.py -------------------------------------------------------------------------------- /projects/torch_vct/utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/projects/torch_vct/utils/schedulers.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/setup.py -------------------------------------------------------------------------------- /tests/cached_data/create_cached_tensorflow_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/cached_data/create_cached_tensorflow_data.py -------------------------------------------------------------------------------- /tests/cached_data/dense_image_warp.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/cached_data/dense_image_warp.pkl -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/data/test_clic_2020_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/data/test_clic_2020_image.py -------------------------------------------------------------------------------- /tests/data/test_clic_2020_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/data/test_clic_2020_video.py -------------------------------------------------------------------------------- /tests/data/test_div2k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/data/test_div2k.py -------------------------------------------------------------------------------- /tests/data/test_kodak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/data/test_kodak.py -------------------------------------------------------------------------------- /tests/data/test_open_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/data/test_open_images.py -------------------------------------------------------------------------------- /tests/data/test_vimeo_90k_septuplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/data/test_vimeo_90k_septuplet.py -------------------------------------------------------------------------------- /tests/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/distributions/__init__.py -------------------------------------------------------------------------------- /tests/distributions/test_noisy_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/distributions/test_noisy_normal.py -------------------------------------------------------------------------------- /tests/distributions/test_uniform_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/distributions/test_uniform_noise.py -------------------------------------------------------------------------------- /tests/functional/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/__init__.py -------------------------------------------------------------------------------- /tests/functional/test_count_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_count_flops.py -------------------------------------------------------------------------------- /tests/functional/test_dense_image_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_dense_image_warp.py -------------------------------------------------------------------------------- /tests/functional/test_estimate_tails.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_estimate_tails.py -------------------------------------------------------------------------------- /tests/functional/test_hsv_to_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_hsv_to_rgb.py -------------------------------------------------------------------------------- /tests/functional/test_image_255_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_image_255_scale.py -------------------------------------------------------------------------------- /tests/functional/test_information_content.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_information_content.py -------------------------------------------------------------------------------- /tests/functional/test_log_cdf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_log_cdf.py -------------------------------------------------------------------------------- /tests/functional/test_log_expm1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_log_expm1.py -------------------------------------------------------------------------------- /tests/functional/test_log_ndtr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_log_ndtr.py -------------------------------------------------------------------------------- /tests/functional/test_log_survival_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_log_survival_function.py -------------------------------------------------------------------------------- /tests/functional/test_lower_bound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_lower_bound.py -------------------------------------------------------------------------------- /tests/functional/test_lower_tail.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_lower_tail.py -------------------------------------------------------------------------------- /tests/functional/test_ndtr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_ndtr.py -------------------------------------------------------------------------------- /tests/functional/test_optical_flow_to_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_optical_flow_to_rgb.py -------------------------------------------------------------------------------- /tests/functional/test_pad_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_pad_image.py -------------------------------------------------------------------------------- /tests/functional/test_quantization_offset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_quantization_offset.py -------------------------------------------------------------------------------- /tests/functional/test_soft_round.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_soft_round.py -------------------------------------------------------------------------------- /tests/functional/test_soft_round_conditional_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_soft_round_conditional_mean.py -------------------------------------------------------------------------------- /tests/functional/test_soft_round_inverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_soft_round_inverse.py -------------------------------------------------------------------------------- /tests/functional/test_survival_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_survival_function.py -------------------------------------------------------------------------------- /tests/functional/test_upper_tail.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/functional/test_upper_tail.py -------------------------------------------------------------------------------- /tests/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/layers/__init__.py -------------------------------------------------------------------------------- /tests/layers/test_channelnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/layers/test_channelnorm.py -------------------------------------------------------------------------------- /tests/layers/test_continuous_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/layers/test_continuous_entropy.py -------------------------------------------------------------------------------- /tests/loss_fn/test_binary_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/loss_fn/test_binary_cross_entropy.py -------------------------------------------------------------------------------- /tests/loss_fn/test_mse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/loss_fn/test_mse_loss.py -------------------------------------------------------------------------------- /tests/loss_fn/test_mse_lpips_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/loss_fn/test_mse_lpips_loss.py -------------------------------------------------------------------------------- /tests/loss_fn/test_oasis_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/loss_fn/test_oasis_loss.py -------------------------------------------------------------------------------- /tests/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/__init__.py -------------------------------------------------------------------------------- /tests/metrics/test_calc_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_calc_psnr.py -------------------------------------------------------------------------------- /tests/metrics/test_dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_dists.py -------------------------------------------------------------------------------- /tests/metrics/test_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_fid.py -------------------------------------------------------------------------------- /tests/metrics/test_kid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_kid.py -------------------------------------------------------------------------------- /tests/metrics/test_msssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_msssim.py -------------------------------------------------------------------------------- /tests/metrics/test_pickle_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_pickle_size.py -------------------------------------------------------------------------------- /tests/metrics/test_swav_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_swav_fid.py -------------------------------------------------------------------------------- /tests/metrics/test_update_patch_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/metrics/test_update_patch_fid.py -------------------------------------------------------------------------------- /tests/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/models/__init__.py -------------------------------------------------------------------------------- /tests/models/test_hific_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/models/test_hific_autoencoder.py -------------------------------------------------------------------------------- /tests/models/test_hific_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/models/test_hific_encoder.py -------------------------------------------------------------------------------- /tests/models/test_hific_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/models/test_hific_generator.py -------------------------------------------------------------------------------- /tests/models/test_unet_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/models/test_unet_discriminator.py -------------------------------------------------------------------------------- /tests/test_layers_gdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/test_layers_gdn.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tests/utils.py -------------------------------------------------------------------------------- /tutorials/Flop_Count_Example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tutorials/Flop_Count_Example.ipynb -------------------------------------------------------------------------------- /tutorials/Metrics_Example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tutorials/Metrics_Example.ipynb -------------------------------------------------------------------------------- /tutorials/image_compression.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/NeuralCompression/HEAD/tutorials/image_compression.ipynb --------------------------------------------------------------------------------