Skip to content

Commit

Permalink
make folders for IEEE Access
Browse files Browse the repository at this point in the history
  • Loading branch information
kktsubota committed Feb 27, 2023
1 parent 34c8375 commit 7fe35ed
Show file tree
Hide file tree
Showing 9 changed files with 55 additions and 49 deletions.
2 changes: 2 additions & 0 deletions Access23/README.md
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.
51 changes: 51 additions & 0 deletions ICIP21/README.md
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.
51 changes: 2 additions & 49 deletions README.md
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}
}
```

0 comments on commit 7fe35ed

Please sign in to comment.