-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
9 changed files
with
55 additions
and
49 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
# Comprehensive Comparisons of Uniform Quantization in Deep Image Compression | ||
Official implementation of "Comprehensive Comparisons of Uniform Quantization in Deep Image Compression" in IEEE Access 2023. |
File renamed without changes.
File renamed without changes.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,51 @@ | ||
# Comprehensive Comparisons of Uniform Quantizers for Deep Image Compression | ||
Official implementation of "Comprehensive Comparisons of Uniform Quantizers for Deep Image Compression" in ICIP 2021. | ||
|
||
## Dataset | ||
Prepare ImageNet and the Kodak dataset. | ||
|
||
We removed images whose shorter edge is smaller than 256 pixels for ImageNet. | ||
|
||
Download the list of images that we used in our experiments by the following commands. | ||
|
||
``` | ||
mkdir datasets | ||
wget https://github.com/kktsubota/uniform-quantizers/releases/download/pre/ImageNet256.txt -O datasets/ImageNet256.txt | ||
``` | ||
|
||
## Environment | ||
* CUDA==10.0 | ||
* CUDNN==7.6.0 | ||
* Python | ||
|
||
``` | ||
pip install pipenv | ||
pipenv install | ||
``` | ||
|
||
Refer to `Pipfile` if you download packages manually. | ||
|
||
## Usage | ||
|
||
```bash | ||
# train a model | ||
python main.py --verbose --checkpoint_dir checkpoints/l0.01_aun_aun --qua_ent AUN-Q train --lambda 0.01 --qua_dec AUN-Q --train_root /path/to/ImageNet/train/ | ||
|
||
# evaluate the model | ||
python evaluate.py /path/to/Kodak/images/ --qua_ent AUN-Q --checkpoint_dir checkpoints/l0.01_aun_aun/ | ||
``` | ||
|
||
You can train other combinations of approximation methods by specifying `--qua_ent` and `--qua_dec`. | ||
|
||
Please select from `{AUN-Q, STE-Q, U-Q, SGA-Q}` for each option. | ||
|
||
## Citation | ||
``` | ||
@inproceedings{tsubotaICIP21, | ||
title = {Comprehensive Comparisons of Uniform Quantizers for Deep Image Compression}, | ||
author = {Tsubota, Koki and Aizawa, Kiyoharu}, | ||
booktitle = {ICIP}, | ||
year = {2021}, | ||
pages={2089-2093} | ||
} | ||
``` |
File renamed without changes.
File renamed without changes.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,50 +1,3 @@ | ||
# uniform-quantizers | ||
Official implementation of "Comprehensive Comparisons of Uniform Quantizers for Deep Image Compression". | ||
# Uniform-Quantizers | ||
This repository provides the source code of our two papers: "Comprehensive Comparisons of Uniform Quantizers for Deep Image Compression" in ICIP 2021 and "Comprehensive Comparisons of Uniform Quantization in Deep Image Compression" in IEEE Access 2023. | ||
|
||
## Dataset | ||
Prepare ImageNet and the Kodak dataset. | ||
|
||
We removed images whose shorter edge is smaller than 256 pixels for ImageNet. | ||
|
||
Download the list of images that we used in our experiments by the following commands. | ||
|
||
``` | ||
mkdir datasets | ||
wget https://github.com/kktsubota/uniform-quantizers/releases/download/pre/ImageNet256.txt -O datasets/ImageNet256.txt | ||
``` | ||
|
||
## Environment | ||
* CUDA==10.0 | ||
* CUDNN==7.6.0 | ||
* Python | ||
|
||
``` | ||
pip install pipenv | ||
pipenv install | ||
``` | ||
|
||
Refer to `Pipfile` if you download packages manually. | ||
|
||
## Usage | ||
|
||
```bash | ||
# train a model | ||
python main.py --verbose --checkpoint_dir checkpoints/l0.01_aun_aun --qua_ent AUN-Q train --lambda 0.01 --qua_dec AUN-Q --train_root /path/to/ImageNet/train/ | ||
|
||
# evaluate the model | ||
python evaluate.py /path/to/Kodak/images/ --qua_ent AUN-Q --checkpoint_dir checkpoints/l0.01_aun_aun/ | ||
``` | ||
|
||
You can train other combinations of approximation methods by specifying `--qua_ent` and `--qua_dec`. | ||
|
||
Please select from `{AUN-Q, STE-Q, U-Q, SGA-Q}` for each option. | ||
|
||
## Citation | ||
``` | ||
@inproceedings{tsubotaICIP21, | ||
title = {Comprehensive Comparisons of Uniform Quantizers for Deep Image Compression}, | ||
author = {Tsubota, Koki and Aizawa, Kiyoharu}, | ||
booktitle = {ICIP}, | ||
year = {2021} | ||
} | ||
``` |