Hugging Face's logo

Phips
/
BHI_Filtering_Post

TensorBoard
Model card Files Metrics Community
Phips commited on
Commit
34f3942
•
1 Parent(s): 93d0494

restructure

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. df2k/dataset/DF2K/DF2K_HR.zip +3 -0
  2. df2k/dataset/DF2K/DF2K_LR.zip +3 -0
  3. df2k/dataset/scores/DF2K_HR_blockiness_score.txt +0 -0
  4. df2k/dataset/scores/DF2K_HR_hyperiqa_score.txt +0 -0
  5. df2k/dataset/scores/DF2K_HR_ic9600_score.txt +0 -0
  6. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/4xDF2K_plksr_tiny.toml +57 -0
  7. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/models/net_g_500000.pth +3 -0
  8. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/train_4xDF2K_plksr_tiny_20240926_193309.log +0 -0
  9. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/training_states/500000.state +3 -0
  10. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/4xDF2K_plksr_tiny_bf16.toml +59 -0
  11. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/models/net_g_200000.pth +3 -0
  12. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/train_4xDF2K_plksr_tiny_bf16_20241001_035536.log +0 -0
  13. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/training_states/200000.state +3 -0
  14. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/4xDF2K_plksr_tiny_fp16.toml +58 -0
  15. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_100000.pth +3 -0
  16. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_200000.pth +3 -0
  17. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/train_4xDF2K_plksr_tiny_fp16_20241001_002907.log +0 -0
  18. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/100000.state +3 -0
  19. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/200000.state +3 -0
  20. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/4xDF2K_plksr_tiny_fp16_full.toml +58 -0
  21. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/models/net_g_500000.pth +3 -0
  22. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/train_4xDF2K_plksr_tiny_fp16_full_20241009_171731.log +0 -0
  23. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/training_states/500000.state +3 -0
  24. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/4xDF2K_plksr_tiny_fp16_multiscaled.toml +58 -0
  25. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/models/net_g_200000.pth +3 -0
  26. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/train_4xDF2K_plksr_tiny_fp16_mutliscaled_20241002_000753.log +0 -0
  27. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/training_states/200000.state +3 -0
  28. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/4xDF2K_plksr_tiny_fp16_patch64.toml +58 -0
  29. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/models/net_g_200000.pth +3 -0
  30. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_033408.log +0 -0
  31. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_082607.log +0 -0
  32. df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/training_states/200000.state +3 -0
  33. df2k/models/baseline/configs/4xDF2K_plksr_tiny.toml +57 -0
  34. df2k/models/baseline/configs/4xDF2K_plksr_tiny_bf16.toml +59 -0
  35. df2k/models/baseline/configs/4xDF2K_plksr_tiny_fp16.toml +58 -0
  36. df2k/models/curated/models_500k/4xDF2K_curated_plksr_tiny_fp16_full_net_g_500000.pth +3 -0
  37. df2k/models/curated/models_500k/4xDF2K_plksr_tiny_fp16_full_net_g_500000.pth +3 -0
  38. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/4xDF2K_curated_plksr_tiny_fp16_full.toml +62 -0
  39. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_10000.pth +3 -0
  40. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_100000.pth +3 -0
  41. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_110000.pth +3 -0
  42. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_120000.pth +3 -0
  43. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_130000.pth +3 -0
  44. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_140000.pth +3 -0
  45. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_150000.pth +3 -0
  46. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_160000.pth +3 -0
  47. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_170000.pth +3 -0
  48. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_180000.pth +3 -0
  49. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_190000.pth +3 -0
  50. df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_20000.pth +3 -0
df2k/dataset/DF2K/DF2K_HR.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:616a6395c7565b42cc3cd9a6f80e100f49b4f3b314ab7fd004576a171671e3ea
3
+ size 8849389127
df2k/dataset/DF2K/DF2K_LR.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9aa75f886d16bc60e4be603224af6b2c6434a65c4fba3fc5a05a6e189ac5f488
3
+ size 755456013
df2k/dataset/scores/DF2K_HR_blockiness_score.txt ADDED
The diff for this file is too large to render. See raw diff
 
df2k/dataset/scores/DF2K_HR_hyperiqa_score.txt ADDED
The diff for this file is too large to render. See raw diff
 
df2k/dataset/scores/DF2K_HR_ic9600_score.txt ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/4xDF2K_plksr_tiny.toml ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny"
2
+ model_type = "image"
3
+ scale = 4
4
+ manual_seed = 1024
5
+
6
+ [datasets.train]
7
+ type = "paired"
8
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
9
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
10
+ patch_size = 32
11
+ batch_size = 16
12
+
13
+ [datasets.val]
14
+ name = "val"
15
+ type = "paired"
16
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
17
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
18
+ [val]
19
+ val_freq = 10000
20
+ save_img = false
21
+ [val.metrics.psnr]
22
+ type = "calculate_psnr"
23
+ [val.metrics.ssim]
24
+ type = "calculate_ssim"
25
+ [val.metrics.dists]
26
+ type = "calculate_dists"
27
+ better = "lower"
28
+
29
+ [path]
30
+
31
+ [network_g]
32
+ #type = "plksr"
33
+ type = "plksr_tiny"
34
+
35
+ [train]
36
+ ema = 0.999
37
+
38
+ [train.optim_g]
39
+ type = "adamw"
40
+ lr = 1e-4
41
+ betas = [ 0.9, 0.99 ]
42
+ weight_decay = 0
43
+
44
+ [train.scheduler]
45
+ type = "multisteplr"
46
+ milestones = [ 60000, 120000 ]
47
+ gamma = 0.5
48
+
49
+ [train.pixel_opt]
50
+ type = "L1Loss"
51
+ loss_weight = 1.0
52
+ reduction = "mean"
53
+
54
+ [logger]
55
+ total_iter = 500000
56
+ save_checkpoint_freq = 10000
57
+ use_tb_logger = true
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/models/net_g_500000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccfef467d9603724f3bcbf938ad01d7560ef162421aeaa9dfea183c9b7a1a676
3
+ size 9513138
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/train_4xDF2K_plksr_tiny_20240926_193309.log ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny/training_states/500000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de125d6476b28e0f8f500d3f8d6dd38d8399d8836c92bfa3f8673e0172f9a671
3
+ size 19045418
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/4xDF2K_plksr_tiny_bf16.toml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_bf16"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ bfloat16 = true
6
+ manual_seed = 1024
7
+
8
+ [datasets.train]
9
+ type = "paired"
10
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
11
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
12
+ patch_size = 32
13
+ batch_size = 16
14
+
15
+ [datasets.val]
16
+ name = "val"
17
+ type = "paired"
18
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
19
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
20
+ [val]
21
+ val_freq = 10000
22
+ save_img = false
23
+ [val.metrics.psnr]
24
+ type = "calculate_psnr"
25
+ [val.metrics.ssim]
26
+ type = "calculate_ssim"
27
+ [val.metrics.dists]
28
+ type = "calculate_dists"
29
+ better = "lower"
30
+
31
+ [path]
32
+
33
+ [network_g]
34
+ #type = "plksr"
35
+ type = "plksr_tiny"
36
+
37
+ [train]
38
+ ema = 0.999
39
+
40
+ [train.optim_g]
41
+ type = "adamw"
42
+ lr = 1e-4
43
+ betas = [ 0.9, 0.99 ]
44
+ weight_decay = 0
45
+
46
+ [train.scheduler]
47
+ type = "multisteplr"
48
+ milestones = [ 60000, 120000 ]
49
+ gamma = 0.5
50
+
51
+ [train.pixel_opt]
52
+ type = "L1Loss"
53
+ loss_weight = 1.0
54
+ reduction = "mean"
55
+
56
+ [logger]
57
+ total_iter = 200000
58
+ save_checkpoint_freq = 10000
59
+ use_tb_logger = true
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df446e3520a58cb039fd7c107e991cc2faf1f4ef6e05bc0e2fb3e4231cb320ad
3
+ size 9513138
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/train_4xDF2K_plksr_tiny_bf16_20241001_035536.log ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_bf16/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c9af2c4e23b026ae8963f802ff61b0a13a38fc54b8dec90a3656822605733ae
3
+ size 19045418
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/4xDF2K_plksr_tiny_fp16.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
11
+ patch_size = 32
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 200000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_100000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd0e027df7bc6e3b7c5b901f1aa1e41130fc5b8b796cf9ad1b73b9c9af0c62f3
3
+ size 9513138
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6d1560850df99e656c4cf04e15a00fc4e74ed9d3e4ea11b7f24ff7a4227f872
3
+ size 9513138
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/train_4xDF2K_plksr_tiny_fp16_20241001_002907.log ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/100000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f97e9d96d051ab4723ffcca5a3ef224465d6d3e0781670487f6de47a3383779d
3
+ size 19045418
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:854d12454ba1139b511da43cf1c310f3b77139f7c4d51169a9de8942efe698cf
3
+ size 19045418
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/4xDF2K_plksr_tiny_fp16_full.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16_full"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
11
+ patch_size = 32
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 500000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/models/net_g_500000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b3147f206c25a52fd922ad8cfc85be8da3cc760ad5ac135e70444cb701b0683
3
+ size 9513138
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/train_4xDF2K_plksr_tiny_fp16_full_20241009_171731.log ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_full/training_states/500000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:306638a02f6869135eaf7298409d4c310c08d717c83a3c65e73ccac50191a9a2
3
+ size 19045418
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/4xDF2K_plksr_tiny_fp16_multiscaled.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16_mutliscaled"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_multiscaled_tiled_merged_HR/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_multiscaled_tiled_merged_LR/'
11
+ patch_size = 32
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 200000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9cbb26b3404ece32a1ebcd163d56502644105846a11cf72db20acbfe799d205
3
+ size 9513138
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/train_4xDF2K_plksr_tiny_fp16_mutliscaled_20241002_000753.log ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_mutliscaled/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63f41bb98712163a9b6fdf51192cb1c0f16f3bfcd0450cd6daa0a803ffc706c4
3
+ size 19045418
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/4xDF2K_plksr_tiny_fp16_patch64.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16_patch64"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
11
+ patch_size = 64
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 200000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/models/net_g_200000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:682f0143c2c8c2d36e608591ff43436bde79a161d90c26aff6f76625a3d98d9a
3
+ size 9513138
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_033408.log ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/train_4xDF2K_plksr_tiny_fp16_patch64_20241002_082607.log ADDED
The diff for this file is too large to render. See raw diff
 
df2k/models/baseline/DF2K/4xDF2K_plksr_tiny_fp16_patch64/training_states/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a63a9ae8e1983372f0bffd02176928db53106bb9bda152c300d11fff68a15522
3
+ size 19045418
df2k/models/baseline/configs/4xDF2K_plksr_tiny.toml ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny"
2
+ model_type = "image"
3
+ scale = 4
4
+ manual_seed = 1024
5
+
6
+ [datasets.train]
7
+ type = "paired"
8
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
9
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
10
+ patch_size = 32
11
+ batch_size = 16
12
+
13
+ [datasets.val]
14
+ name = "val"
15
+ type = "paired"
16
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
17
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
18
+ [val]
19
+ val_freq = 10000
20
+ save_img = false
21
+ [val.metrics.psnr]
22
+ type = "calculate_psnr"
23
+ [val.metrics.ssim]
24
+ type = "calculate_ssim"
25
+ [val.metrics.dists]
26
+ type = "calculate_dists"
27
+ better = "lower"
28
+
29
+ [path]
30
+
31
+ [network_g]
32
+ #type = "plksr"
33
+ type = "plksr_tiny"
34
+
35
+ [train]
36
+ ema = 0.999
37
+
38
+ [train.optim_g]
39
+ type = "adamw"
40
+ lr = 1e-4
41
+ betas = [ 0.9, 0.99 ]
42
+ weight_decay = 0
43
+
44
+ [train.scheduler]
45
+ type = "multisteplr"
46
+ milestones = [ 60000, 120000 ]
47
+ gamma = 0.5
48
+
49
+ [train.pixel_opt]
50
+ type = "L1Loss"
51
+ loss_weight = 1.0
52
+ reduction = "mean"
53
+
54
+ [logger]
55
+ total_iter = 1000000
56
+ save_checkpoint_freq = 10000
57
+ use_tb_logger = true
df2k/models/baseline/configs/4xDF2K_plksr_tiny_bf16.toml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_bf16"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ bfloat16 = true
6
+ manual_seed = 1024
7
+
8
+ [datasets.train]
9
+ type = "paired"
10
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
11
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
12
+ patch_size = 32
13
+ batch_size = 16
14
+
15
+ [datasets.val]
16
+ name = "val"
17
+ type = "paired"
18
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
19
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
20
+ [val]
21
+ val_freq = 10000
22
+ save_img = false
23
+ [val.metrics.psnr]
24
+ type = "calculate_psnr"
25
+ [val.metrics.ssim]
26
+ type = "calculate_ssim"
27
+ [val.metrics.dists]
28
+ type = "calculate_dists"
29
+ better = "lower"
30
+
31
+ [path]
32
+
33
+ [network_g]
34
+ #type = "plksr"
35
+ type = "plksr_tiny"
36
+
37
+ [train]
38
+ ema = 0.999
39
+
40
+ [train.optim_g]
41
+ type = "adamw"
42
+ lr = 1e-4
43
+ betas = [ 0.9, 0.99 ]
44
+ weight_decay = 0
45
+
46
+ [train.scheduler]
47
+ type = "multisteplr"
48
+ milestones = [ 60000, 120000 ]
49
+ gamma = 0.5
50
+
51
+ [train.pixel_opt]
52
+ type = "L1Loss"
53
+ loss_weight = 1.0
54
+ reduction = "mean"
55
+
56
+ [logger]
57
+ total_iter = 200000
58
+ save_checkpoint_freq = 10000
59
+ use_tb_logger = true
df2k/models/baseline/configs/4xDF2K_plksr_tiny_fp16.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name = "4xDF2K_plksr_tiny_fp16"
2
+ model_type = "image"
3
+ scale = 4
4
+ use_amp = true
5
+ manual_seed = 1024
6
+
7
+ [datasets.train]
8
+ type = "paired"
9
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_HR_tiled/'
10
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_LR_tiled/'
11
+ patch_size = 32
12
+ batch_size = 16
13
+
14
+ [datasets.val]
15
+ name = "val"
16
+ type = "paired"
17
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
18
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
19
+ [val]
20
+ val_freq = 10000
21
+ save_img = false
22
+ [val.metrics.psnr]
23
+ type = "calculate_psnr"
24
+ [val.metrics.ssim]
25
+ type = "calculate_ssim"
26
+ [val.metrics.dists]
27
+ type = "calculate_dists"
28
+ better = "lower"
29
+
30
+ [path]
31
+
32
+ [network_g]
33
+ #type = "plksr"
34
+ type = "plksr_tiny"
35
+
36
+ [train]
37
+ ema = 0.999
38
+
39
+ [train.optim_g]
40
+ type = "adamw"
41
+ lr = 1e-4
42
+ betas = [ 0.9, 0.99 ]
43
+ weight_decay = 0
44
+
45
+ [train.scheduler]
46
+ type = "multisteplr"
47
+ milestones = [ 60000, 120000 ]
48
+ gamma = 0.5
49
+
50
+ [train.pixel_opt]
51
+ type = "L1Loss"
52
+ loss_weight = 1.0
53
+ reduction = "mean"
54
+
55
+ [logger]
56
+ total_iter = 200000
57
+ save_checkpoint_freq = 10000
58
+ use_tb_logger = true
df2k/models/curated/models_500k/4xDF2K_curated_plksr_tiny_fp16_full_net_g_500000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:374be0924bdc9e11c44555fa68b419d821c49621e4c1f7c904f36a9520343a85
3
+ size 9513138
df2k/models/curated/models_500k/4xDF2K_plksr_tiny_fp16_full_net_g_500000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b3147f206c25a52fd922ad8cfc85be8da3cc760ad5ac135e70444cb701b0683
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/4xDF2K_curated_plksr_tiny_fp16_full.toml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # GENERATE TIME: Wed Oct 9 09:06:08 2024
2
+ # CMD:
3
+ # train.py -opt options/plksr_tiny/merged/full_merged/4xDF2K_curated_plksr_tiny_fp16_full.toml
4
+
5
+ name = "4xDF2K_curated_plksr_tiny_fp16_full"
6
+ model_type = "image"
7
+ scale = 4
8
+ use_amp = true
9
+ manual_seed = 1024
10
+
11
+ [datasets.train]
12
+ type = "paired"
13
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_block_30_hyperiqa_02_ic9600_05_HR/'
14
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/train/DF2K_block_30_hyperiqa_02_ic9600_05_LR/'
15
+ patch_size = 32
16
+ batch_size = 16
17
+
18
+ [datasets.val]
19
+ name = "val"
20
+ type = "paired"
21
+ dataroot_gt = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_HR/'
22
+ dataroot_lq = '/home/phips/Documents/Datasets/wip/DF2K/test/DIV2K_valid_LR_bicubic/X4/'
23
+ [val]
24
+ val_freq = 10000
25
+ save_img = false
26
+ [val.metrics.psnr]
27
+ type = "calculate_psnr"
28
+ [val.metrics.ssim]
29
+ type = "calculate_ssim"
30
+ [val.metrics.dists]
31
+ type = "calculate_dists"
32
+ better = "lower"
33
+
34
+ [path]
35
+
36
+ [network_g]
37
+ #type = "plksr"
38
+ type = "plksr_tiny"
39
+
40
+ [train]
41
+ ema = 0.999
42
+
43
+ [train.optim_g]
44
+ type = "adamw"
45
+ lr = 1e-4
46
+ betas = [ 0.9, 0.99 ]
47
+ weight_decay = 0
48
+
49
+ [train.scheduler]
50
+ type = "multisteplr"
51
+ milestones = [ 60000, 120000 ]
52
+ gamma = 0.5
53
+
54
+ [train.pixel_opt]
55
+ type = "L1Loss"
56
+ loss_weight = 1.0
57
+ reduction = "mean"
58
+
59
+ [logger]
60
+ total_iter = 520000
61
+ save_checkpoint_freq = 10000
62
+ use_tb_logger = true
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_10000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57474547f58711693906c7b1fa07826ad7305a8bdd1996b747d5f7a29dc2145c
3
+ size 9513034
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_100000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97fe03919819b5f290ac900b772cbc677934cf7ee2e3cb07e3c115eab6ffb043
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_110000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62498470e202e917f39caa22a8b37eea7c48e7e7bf0220fbd9f321737612ebfe
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_120000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:354e4a6c05e898a19522347f2567908ec627ade8d953f5e1a7d9e4898ad5617e
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_130000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1436c4bbe84c0c98e6a214f5b55086e2ccc857e290a78ffd3a731cbfa4df3f2e
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_140000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec5322b155808eb535f22cd5eabd2a7ba13caf8e4a6920f1a532ebb5dae6b134
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_150000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85e995bd4fabccb2c48e9a39de6f2ddd50a4fea89d7fb0c99e8693eb56f61533
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_160000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74cbc97e952f80d8cc411d50a7a16b511cfa54cf2be1b598f37df875f1bf5a2f
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_170000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f142d2da2e95c7da0de86db203eaf01c1d3666b79dc7accd910c3af5713dcaa6
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_180000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ad2327d062a60fefb03212fe6e1826c5d09eb09545409ef47561d84f74b3098
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_190000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ecc3aa78684ec3633b052a348f25facda66c8fdc707a6df79b604a01b2b25a42
3
+ size 9513138
df2k/models/curated/training/4xDF2K_curated_plksr_tiny_fp16_full/models/net_g_20000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90d99d19c373b682981333f09a663b3fa47d6950f73092d151c1578054d55c5f
3
+ size 9513034