Skip to content

Commit 598d6a9

Browse files
authored
Create README.md
1 parent 2957ccf commit 598d6a9

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

README.md

+33
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
# Autoencoder-Image-Compression
2+
Pytorch implementation for image compression and reconstruction via autoencoder
3+
4+
This is an autoencoder with cylic loss and coding parsing loss for image compression and reconstruction. Network backbone is simple 3-layer
5+
fully conv (encoder) and symmetrical for decoder. Finally it can achieve 21 mean PSNR on CLIC dataset (CVPR 2019 workshop).
6+
7+
You can download
8+
training data from this url: https://drive.google.com/drive/folders/1wU1CO6WcQOraIaY2KSk7cRVaAXcm_A2R?usp=sharing
9+
validation data: https://drive.google.com/drive/folders/113EcrAdcxfVqs8BVt4PZjwUEyVz7VVa-?usp=sharing
10+
11+
Organize your data with this structure:
12+
13+
Data
14+
|
15+
|---train
16+
|
17+
|---image1.xxx
18+
|---image2.xxx
19+
.
20+
.
21+
.
22+
23+
Data_valid
24+
|
25+
|---train
26+
|
27+
|---image1.xxx
28+
|---image2.xxx
29+
.
30+
.
31+
.
32+
33+
You can train your own model via run_train.sh and modify config as your needs. Prediction for the valid data via run_test.sh

0 commit comments

Comments
 (0)