File tree 1 file changed +33
-0
lines changed
1 file changed +33
-0
lines changed Original file line number Diff line number Diff line change
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
You can’t perform that action at this time.
0 commit comments