Name		Name	Last commit message	Last commit date
parent directory ..
datas		datas
models		models
README.md		README.md
train.py		train.py

README.md

Train an autoencoder with SSIM & MS-SSIM

Prepare dataset

Download CLIC datase from http://clic.compression.cc/2021/tasks/index.html.
Unzip them into datasets.

The structure of the directory:

- datasets
  - CLIC
    - train
      - *.png
      - ...
    - valid
      - *.png
      - ...

Train

SSIM loss:
```
$ python train.py --loss_type ssim
```
MS-SSIM loss:
```
$ python train.py --loss_type ms_ssim
```