Initial commit

.vscode/launch.json

@@ -0,0 +1,51 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python: Current File",
+            "type": "python",
+            "request": "launch",
+            "program": "${file}",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        },
+        {
+            "name": "RealESRGAN training",
+            "type": "python",
+            "module": "torch.distributed.launch",
+            "request": "launch",
+            "args": [
+                "--nproc_per_node=4",
+                "--master_port=4321",
+                "realesrgan/train.py",
+                "-opt",
+                "options/train_realesrnet_x4plus.yml",
+                "--launcher",
+                "pytorch",
+                "--debug"
+            ],
+            "cwd": "${workspaceFolder}",
+            "env": {},
+            "console": "integratedTerminal",
+            "justMyCode": false
+        },
+        {
+            "name": "RealESRGAN training",
+            "type": "python",
+            "module": "realesrgan/train.py",
+            "request": "launch",
+            "args": [
+                "-opt",
+                "options/train_realesrnet_x4plus.yml",
+                "--auto_resume"
+            ],
+            "cwd": "${workspaceFolder}",
+            "env": {},
+            "console": "integratedTerminal",
+            "justMyCode": false
+        }
+    ]
+}

options/finetune_realesrgan_x4plus.yml

@@ -36,10 +36,10 @@ queue_size: 180
 # dataset and data loader settings
 datasets:
   train:
-    name: DF2K+OST
+    name: engravings_dataset
     type: RealESRGANDataset
-    dataroot_gt: datasets/DF2K
-    meta_info: datasets/DF2K/meta_info/meta_info_DF2Kmultiscale+OST_sub.txt
+    dataroot_gt: C:\Users\ianko\Desktop\Projects\AbacusBooks\Real-ESRGAN\datasets\engravings_dataset\images
+    meta_info: C:\Users\ianko\Desktop\Projects\AbacusBooks\Real-ESRGAN\datasets\engravings_dataset\meta_info\meta_info_engraving_dataset.txt
     io_backend:
       type: disk
 
@@ -99,10 +99,10 @@ network_d:
 # path
 path:
   # use the pre-trained Real-ESRNet model
-  pretrain_network_g: experiments/pretrained_models/RealESRNet_x4plus.pth
+  pretrain_network_g: C:\Users\ianko\Desktop\Projects\AbacusBooks\Real-ESRGAN\experiments\pretrained_models\RealESRNet_x4plus.pth
   param_key_g: params_ema
   strict_load_g: true
-  pretrain_network_d: experiments/pretrained_models/RealESRGAN_x4plus_netD.pth
+  pretrain_network_d: C:\Users\ianko\Desktop\Projects\AbacusBooks\Real-ESRGAN\experiments/pretrained_models/RealESRGAN_x4plus_netD.pth
   param_key_d: params
   strict_load_d: true
   resume_state: ~

options/train_realesrnet_x4plus.yml

@@ -28,15 +28,15 @@ gray_noise_prob2: 0.4
 jpeg_range2: [30, 95]
 
 gt_size: 256
-queue_size: 180
+queue_size: 200
 
 # dataset and data loader settings
 datasets:
   train:
-    name: DF2K+OST
+    name: engravings_dataset
     type: RealESRGANDataset
-    dataroot_gt: datasets/DF2K
-    meta_info: datasets/DF2K/meta_info/meta_info_DF2Kmultiscale+OST_sub.txt
+    dataroot_gt: C:\Users\ianko\Desktop\Projects\AbacusBooks\Real-ESRGAN\datasets\engravings_dataset\images
+    meta_info: C:\Users\ianko\Desktop\Projects\AbacusBooks\Real-ESRGAN\datasets\engravings_dataset\meta_info\meta_info_engraving_dataset_sub.txt
     io_backend:
       type: disk
 
@@ -65,9 +65,10 @@ datasets:
     # data loader
     use_shuffle: true
     num_worker_per_gpu: 5
-    batch_size_per_gpu: 12
+    batch_size_per_gpu: 4
     dataset_enlarge_ratio: 1
-    prefetch_mode: ~
+    prefetch_mode: 'cuda'
+    pin_memory: True
 
   # Uncomment these for validation
   # val:
@@ -89,7 +90,7 @@ network_g:
 
 # path
 path:
-  pretrain_network_g: experiments/pretrained_models/ESRGAN_SRx4_DF2KOST_official-ff704c30.pth
+  pretrain_network_g: C:\Users\ianko\Desktop\Projects\AbacusBooks\Real-ESRGAN\experiments\pretrained_models\ESRGAN_SRx4_DF2KOST_official-ff704c30.pth
   param_key_g: params_ema
   strict_load_g: true
   resume_state: ~

realesrgan/meta_info/book_engravings.txt

@@ -0,0 +1,39 @@
+https://archive.org/details/elementsofstylei00broo_0/page/1/mode/1up
+https://archive.org/details/cartouches00mite/page/n22/mode/1up
+https://archive.org/details/gothicornamentso00pugi/page/n18/mode/1up
+https://archive.org/details/progressiveameri00crof/page/n44/mode/1up
+https://archive.org/details/designersplanboo00farn/page/n9/mode/1up
+https://archive.org/details/studybookofmediv03king_0/page/n18/mode/1up
+https://archive.org/details/designsforchimni00swan/page/41/mode/1up
+https://archive.org/details/commensuracionpa00arfe/page/145/mode/1up
+https://archive.org/details/gri_33125010570915/page/16/mode/2up
+https://archive.org/details/handbooktocathed12king_0/page/n53/mode/1up
+https://archive.org/details/mittheilungender14kkze/page/n242/mode/1up
+https://archive.org/details/gri_33125006524173/page/n206/mode/1up
+https://archive.org/details/diekonstruktione02uhde/page/206/mode/1up
+https://archive.org/details/observationsonar00rich/page/n111/mode/1up
+https://archive.org/details/regoladellicinqv00vign/page/n28/mode/1up
+https://archive.org/details/samuelismarolois00maro/page/n612/mode/1up
+https://archive.org/details/diekonstruktione41uhde/page/33/mode/1up
+https://archive.org/details/architectvravona00diet/page/21/mode/1up
+https://archive.org/details/detailsofgothica02coll_0/page/n62/mode/1up
+https://archive.org/details/detailsofgothica01coll_0/page/n230/mode/1up
+https://archive.org/details/mittheilungender14kkze/page/n234/mode/1up
+https://archive.org/details/commensuracionpa00arfe/page/150/mode/1up
+https://archive.org/details/allgemeinebaukon01brey/page/563/mode/1up
+https://archive.org/details/imaginifavolosen00pitt/page/n62/mode/1up
+https://archive.org/details/gri_33125006524173/page/n301/mode/1up
+https://archive.org/details/handbooktocathed41king_0/page/n7/mode/1up
+https://archive.org/details/regoladellicinqv00vign/page/n2/mode/1up
+https://archive.org/details/lesmonumentsarab00marc/page/121/mode/thumb
+https://archive.org/details/analysisoforname00worn_1/page/9/mode/1up
+https://archive.org/details/carpentrybuildin5188unse/page/173/mode/1up
+https://archive.org/details/architectvravona00diet/page/31/mode/1up
+https://archive.org/details/architectvravona00diet/page/32/mode/1up
+https://archive.org/details/detailsofgothica02coll_0/page/n206/mode/1up
+https://archive.org/details/ordinidarchitett00vign/page/n52/mode/1up
+https://archive.org/details/gri_33125012247702/page/n495/mode/2up
+https://archive.org/details/detailsofgothica02coll_0/page/n274/mode/1up
+https://archive.org/details/ornementdeducerc00andr/page/n31/mode/1up
+https://archive.org/details/architectvravonn00meye/page/n89/mode/2up
+https://archive.org/details/diebibelinbilder00schn/page/99/mode/1up