Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
yehua
75b63d3367
|
2 years ago | |
|---|---|---|
| .. | ||
| models | 2 years ago | |
| test | 2 years ago | |
| .gitignore | 2 years ago | |
| 400_iter_19762.pth.ckpt | 2 years ago | |
| 800_iter_19762.pth.ckpt | 2 years ago | |
| 1500_iter_19762.pth.ckpt | 2 years ago | |
| 3000_iter_19844.pth.ckpt | 2 years ago | |
| LICENSE | 2 years ago | |
| Meter.py | 2 years ago | |
| README.md | 2 years ago | |
| __init__.py | 2 years ago | |
| datasets.py | 2 years ago | |
| e2e_gdn_mindspore_lambda400.csv | 2 years ago | |
| e2e_gdn_mindspore_lambda800.csv | 2 years ago | |
| e2e_gdn_mindspore_lambda1500.csv | 2 years ago | |
| e2e_gdn_mindspore_lambda3000.csv | 2 years ago | |
| e2e_pretrain.ckpt | 2 years ago | |
| metric.py | 2 years ago | |
| mindspore_eval.jpg | 2 years ago | |
| model.py | 2 years ago | |
| msssim.py | 2 years ago | |
| sea.jpg | 2 years ago | |
| test_ms-all.py | 2 years ago | |
| test_ms.py | 2 years ago | |
| train.py | 2 years ago | |
README.md
Pytorch Reimplementation for End-to-end Optimized Image Compression
Running Script
CUDA_VISIBLE_DEVICES=0 python train.py --config examples/example/config.json -n baseline
Data Preparation
We need to first prepare the training and validation data.
The trainging data is from flicker.com.
You can obtain the training data according to description of CompressionData
The training details is similar to our another repo compression
No Description
Python CSV Shell
