diff --git a/checkpoints/ROMA_UNSB_001/latest_net_D_ViT.pth b/checkpoints/ROMA_UNSB_001/latest_net_D_ViT.pth new file mode 100644 index 0000000..3f3ffca Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/latest_net_D_ViT.pth differ diff --git a/checkpoints/ROMA_UNSB_001/latest_net_E.pth b/checkpoints/ROMA_UNSB_001/latest_net_E.pth new file mode 100644 index 0000000..1b1ea56 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/latest_net_E.pth differ diff --git a/checkpoints/ROMA_UNSB_001/latest_net_G.pth b/checkpoints/ROMA_UNSB_001/latest_net_G.pth new file mode 100644 index 0000000..7b2e4de Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/latest_net_G.pth differ diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt index bb13a7d..21cf024 100644 --- a/checkpoints/ROMA_UNSB_001/loss_log.txt +++ b/checkpoints/ROMA_UNSB_001/loss_log.txt @@ -81,3 +81,69 @@ (epoch: 1, iters: 700, time: 0.202, data: 0.004) G_GAN: 3.999 D_real_ViT: 0.273 D_fake_ViT: 0.232 G: 4.461 SB: 0.058 (epoch: 1, iters: 800, time: 0.204, data: 0.003) G_GAN: 3.754 D_real_ViT: 0.233 D_fake_ViT: 0.241 G: 4.065 SB: 0.054 (epoch: 1, iters: 900, time: 0.205, data: 0.003) G_GAN: 4.083 D_real_ViT: 0.249 D_fake_ViT: 0.207 G: 4.281 SB: 0.059 +(epoch: 1, iters: 1000, time: 0.206, data: 0.003) G_GAN: 4.079 D_real_ViT: 0.244 D_fake_ViT: 0.207 G: 4.516 SB: 0.062 +(epoch: 1, iters: 1100, time: 0.205, data: 0.005) G_GAN: 3.630 D_real_ViT: 0.212 D_fake_ViT: 0.248 G: 3.740 SB: 0.062 +(epoch: 1, iters: 1200, time: 0.207, data: 0.004) G_GAN: 4.639 D_real_ViT: 0.174 D_fake_ViT: 0.159 G: 4.870 SB: 0.058 +(epoch: 1, iters: 1300, time: 0.208, data: 0.004) G_GAN: 3.868 D_real_ViT: 0.224 D_fake_ViT: 0.198 G: 4.078 SB: 0.078 +(epoch: 1, iters: 1400, time: 0.206, data: 0.003) G_GAN: 3.604 D_real_ViT: 0.170 D_fake_ViT: 0.215 G: 3.819 SB: 0.058 +(epoch: 1, iters: 1500, time: 0.206, data: 0.003) G_GAN: 3.718 D_real_ViT: 0.156 D_fake_ViT: 0.199 G: 4.209 SB: 0.057 +(epoch: 1, iters: 1600, time: 0.206, data: 0.002) G_GAN: 4.552 D_real_ViT: 0.184 D_fake_ViT: 0.142 G: 4.772 SB: 0.054 +(epoch: 1, iters: 1700, time: 0.205, data: 0.003) G_GAN: 3.300 D_real_ViT: 0.189 D_fake_ViT: 0.255 G: 3.759 SB: 0.076 +(epoch: 1, iters: 1800, time: 0.204, data: 0.004) G_GAN: 3.840 D_real_ViT: 0.193 D_fake_ViT: 0.185 G: 4.465 SB: 0.073 +(epoch: 1, iters: 1900, time: 0.204, data: 0.002) G_GAN: 3.524 D_real_ViT: 0.155 D_fake_ViT: 0.207 G: 4.182 SB: 0.053 +(epoch: 1, iters: 2000, time: 0.204, data: 0.003) G_GAN: 3.437 D_real_ViT: 0.169 D_fake_ViT: 0.223 G: 3.529 SB: 0.061 +(epoch: 1, iters: 2100, time: 0.204, data: 0.003) G_GAN: 4.786 D_real_ViT: 0.153 D_fake_ViT: 0.132 G: 5.162 SB: 0.061 +(epoch: 2, iters: 95, time: 0.203, data: 0.003) G_GAN: 4.433 D_real_ViT: 0.192 D_fake_ViT: 0.147 G: 4.737 SB: 0.054 +(epoch: 2, iters: 195, time: 0.204, data: 0.004) G_GAN: 3.592 D_real_ViT: 0.182 D_fake_ViT: 0.199 G: 3.722 SB: 0.054 +(epoch: 2, iters: 295, time: 0.204, data: 0.003) G_GAN: 4.473 D_real_ViT: 0.238 D_fake_ViT: 0.133 G: 4.573 SB: 0.062 +(epoch: 2, iters: 395, time: 0.203, data: 0.003) G_GAN: 3.919 D_real_ViT: 0.175 D_fake_ViT: 0.172 G: 4.029 SB: 0.076 +(epoch: 2, iters: 495, time: 0.204, data: 0.003) G_GAN: 3.096 D_real_ViT: 0.204 D_fake_ViT: 0.234 G: 3.726 SB: 0.054 +(epoch: 2, iters: 595, time: 0.202, data: 0.005) G_GAN: 2.927 D_real_ViT: 0.168 D_fake_ViT: 0.254 G: 3.603 SB: 0.059 +(epoch: 2, iters: 695, time: 0.204, data: 0.003) G_GAN: 3.383 D_real_ViT: 0.198 D_fake_ViT: 0.209 G: 3.787 SB: 0.054 +(epoch: 2, iters: 795, time: 0.205, data: 0.003) G_GAN: 3.755 D_real_ViT: 0.198 D_fake_ViT: 0.185 G: 4.026 SB: 0.075 +(epoch: 2, iters: 895, time: 0.205, data: 0.006) G_GAN: 3.717 D_real_ViT: 0.201 D_fake_ViT: 0.170 G: 3.863 SB: 0.062 +(epoch: 2, iters: 995, time: 0.208, data: 0.003) G_GAN: 3.538 D_real_ViT: 0.268 D_fake_ViT: 0.182 G: 4.136 SB: 0.054 +(epoch: 2, iters: 1095, time: 0.207, data: 0.003) G_GAN: 3.946 D_real_ViT: 0.173 D_fake_ViT: 0.166 G: 4.251 SB: 0.058 +(epoch: 2, iters: 1195, time: 0.206, data: 0.003) G_GAN: 4.461 D_real_ViT: 0.162 D_fake_ViT: 0.136 G: 4.591 SB: 0.058 +(epoch: 2, iters: 1295, time: 0.205, data: 0.005) G_GAN: 3.981 D_real_ViT: 0.132 D_fake_ViT: 0.158 G: 4.403 SB: 0.054 +(epoch: 2, iters: 1395, time: 0.206, data: 0.004) G_GAN: 3.391 D_real_ViT: 0.155 D_fake_ViT: 0.190 G: 3.982 SB: 0.078 +(epoch: 2, iters: 1495, time: 0.206, data: 0.005) G_GAN: 3.082 D_real_ViT: 0.264 D_fake_ViT: 0.212 G: 3.343 SB: 0.061 +(epoch: 2, iters: 1595, time: 0.206, data: 0.003) G_GAN: 4.052 D_real_ViT: 0.149 D_fake_ViT: 0.153 G: 4.585 SB: 0.058 +(epoch: 2, iters: 1695, time: 0.206, data: 0.003) G_GAN: 4.096 D_real_ViT: 0.142 D_fake_ViT: 0.153 G: 4.442 SB: 0.058 +(epoch: 2, iters: 1795, time: 0.206, data: 0.003) G_GAN: 2.935 D_real_ViT: 0.337 D_fake_ViT: 0.258 G: 3.227 SB: 0.061 +(epoch: 2, iters: 1895, time: 0.205, data: 0.003) G_GAN: 3.350 D_real_ViT: 0.149 D_fake_ViT: 0.214 G: 3.626 SB: 0.057 +(epoch: 2, iters: 1995, time: 0.205, data: 0.003) G_GAN: 5.271 D_real_ViT: 0.147 D_fake_ViT: 0.101 G: 5.485 SB: 0.073 +(epoch: 2, iters: 2095, time: 0.209, data: 0.003) G_GAN: 4.324 D_real_ViT: 0.171 D_fake_ViT: 0.129 G: 4.600 SB: 0.054 +(epoch: 3, iters: 90, time: 0.208, data: 0.003) G_GAN: 3.161 D_real_ViT: 0.242 D_fake_ViT: 0.206 G: 3.435 SB: 0.055 +(epoch: 3, iters: 190, time: 0.207, data: 0.003) G_GAN: 4.150 D_real_ViT: 0.264 D_fake_ViT: 0.143 G: 4.741 SB: 0.056 +(epoch: 3, iters: 290, time: 0.207, data: 0.003) G_GAN: 5.194 D_real_ViT: 0.268 D_fake_ViT: 0.096 G: 5.567 SB: 0.054 +(epoch: 3, iters: 390, time: 0.208, data: 0.004) G_GAN: 4.214 D_real_ViT: 0.126 D_fake_ViT: 0.145 G: 4.984 SB: 0.061 +(epoch: 3, iters: 490, time: 0.209, data: 0.003) G_GAN: 4.436 D_real_ViT: 0.283 D_fake_ViT: 0.137 G: 4.779 SB: 0.057 +(epoch: 3, iters: 590, time: 0.208, data: 0.003) G_GAN: 4.017 D_real_ViT: 0.162 D_fake_ViT: 0.172 G: 4.254 SB: 0.061 +(epoch: 3, iters: 690, time: 0.210, data: 0.006) G_GAN: 3.531 D_real_ViT: 0.100 D_fake_ViT: 0.179 G: 3.961 SB: 0.062 +(epoch: 3, iters: 790, time: 0.210, data: 0.003) G_GAN: 3.909 D_real_ViT: 0.325 D_fake_ViT: 0.154 G: 4.414 SB: 0.073 +(epoch: 3, iters: 890, time: 0.209, data: 0.006) G_GAN: 5.494 D_real_ViT: 0.221 D_fake_ViT: 0.082 G: 6.089 SB: 0.072 +(epoch: 3, iters: 990, time: 0.207, data: 0.003) G_GAN: 4.783 D_real_ViT: 0.165 D_fake_ViT: 0.111 G: 5.441 SB: 0.057 +(epoch: 3, iters: 1090, time: 0.207, data: 0.004) G_GAN: 3.759 D_real_ViT: 0.335 D_fake_ViT: 0.166 G: 4.676 SB: 0.056 +(epoch: 3, iters: 1190, time: 0.207, data: 0.005) G_GAN: 3.415 D_real_ViT: 0.166 D_fake_ViT: 0.193 G: 4.474 SB: 0.073 +(epoch: 3, iters: 1290, time: 0.206, data: 0.003) G_GAN: 3.669 D_real_ViT: 0.204 D_fake_ViT: 0.169 G: 4.728 SB: 0.060 +(epoch: 3, iters: 1390, time: 0.205, data: 0.004) G_GAN: 3.358 D_real_ViT: 0.199 D_fake_ViT: 0.187 G: 4.412 SB: 0.055 +(epoch: 3, iters: 1490, time: 0.206, data: 0.003) G_GAN: 3.018 D_real_ViT: 0.302 D_fake_ViT: 0.235 G: 4.051 SB: 0.055 +(epoch: 3, iters: 1590, time: 0.206, data: 0.005) G_GAN: 4.606 D_real_ViT: 0.168 D_fake_ViT: 0.138 G: 5.547 SB: 0.061 +(epoch: 3, iters: 1690, time: 0.205, data: 0.003) G_GAN: 3.548 D_real_ViT: 0.269 D_fake_ViT: 0.195 G: 4.058 SB: 0.071 +(epoch: 3, iters: 1790, time: 0.205, data: 0.003) G_GAN: 3.814 D_real_ViT: 0.201 D_fake_ViT: 0.182 G: 4.869 SB: 0.057 +(epoch: 3, iters: 1890, time: 0.207, data: 0.003) G_GAN: 3.740 D_real_ViT: 0.238 D_fake_ViT: 0.173 G: 4.679 SB: 0.058 +(epoch: 3, iters: 1990, time: 0.206, data: 0.004) G_GAN: 4.623 D_real_ViT: 0.227 D_fake_ViT: 0.129 G: 4.969 SB: 0.071 +(epoch: 3, iters: 2090, time: 0.206, data: 0.003) G_GAN: 3.498 D_real_ViT: 0.254 D_fake_ViT: 0.183 G: 4.168 SB: 0.077 +(epoch: 4, iters: 85, time: 0.206, data: 0.003) G_GAN: 3.626 D_real_ViT: 0.252 D_fake_ViT: 0.167 G: 4.710 SB: 0.055 +(epoch: 4, iters: 185, time: 0.207, data: 0.003) G_GAN: 4.156 D_real_ViT: 0.266 D_fake_ViT: 0.140 G: 4.893 SB: 0.057 +(epoch: 4, iters: 285, time: 0.206, data: 0.003) G_GAN: 3.351 D_real_ViT: 0.233 D_fake_ViT: 0.192 G: 4.369 SB: 0.062 +(epoch: 4, iters: 385, time: 0.205, data: 0.004) G_GAN: 3.621 D_real_ViT: 0.296 D_fake_ViT: 0.157 G: 4.613 SB: 0.077 +(epoch: 4, iters: 485, time: 0.207, data: 0.003) G_GAN: 3.686 D_real_ViT: 0.259 D_fake_ViT: 0.170 G: 4.029 SB: 0.072 +(epoch: 4, iters: 585, time: 0.207, data: 0.003) G_GAN: 4.395 D_real_ViT: 0.075 D_fake_ViT: 0.131 G: 4.856 SB: 0.056 +(epoch: 4, iters: 685, time: 0.207, data: 0.003) G_GAN: 4.882 D_real_ViT: 0.104 D_fake_ViT: 0.108 G: 5.622 SB: 0.054 +(epoch: 4, iters: 785, time: 0.205, data: 0.003) G_GAN: 3.992 D_real_ViT: 0.254 D_fake_ViT: 0.134 G: 4.976 SB: 0.075 +(epoch: 4, iters: 885, time: 0.206, data: 0.006) G_GAN: 4.044 D_real_ViT: 0.187 D_fake_ViT: 0.143 G: 5.105 SB: 0.058 +(epoch: 4, iters: 985, time: 0.206, data: 0.002) G_GAN: 3.378 D_real_ViT: 0.065 D_fake_ViT: 0.195 G: 3.964 SB: 0.062 +(epoch: 4, iters: 1085, time: 0.207, data: 0.003) G_GAN: 3.711 D_real_ViT: 0.184 D_fake_ViT: 0.172 G: 4.842 SB: 0.077 +(epoch: 4, iters: 1185, time: 0.207, data: 0.003) G_GAN: 3.574 D_real_ViT: 0.120 D_fake_ViT: 0.186 G: 3.918 SB: 0.070 diff --git a/checkpoints/ROMA_UNSB_001/train_opt.txt b/checkpoints/ROMA_UNSB_001/train_opt.txt index 8f10014..5124342 100644 --- a/checkpoints/ROMA_UNSB_001/train_opt.txt +++ b/checkpoints/ROMA_UNSB_001/train_opt.txt @@ -29,7 +29,7 @@ input_nc: 3 isTrain: True [default: None] lambda_D_ViT: 1.0 - lambda_GAN: 8.0 [default: 1.0] + lambda_GAN: 1.0 lambda_NCE: 8.0 [default: 1.0] lambda_SB: 1.0 [default: 0.1] lambda_ctn: 1.0 diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png index 09c5f95..317cfba 100644 Binary files a/checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png index c59b80f..f61b3f9 100644 Binary files a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png index e6eb10c..b036139 100644 Binary files a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png index cd56419..1dee2aa 100644 Binary files a/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png and b/checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch002_fake_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch002_fake_B0.png new file mode 100644 index 0000000..fb75cfa Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch002_fake_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_A0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_A0.png new file mode 100644 index 0000000..3e57a11 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_A0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_A_noisy.png b/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_A_noisy.png new file mode 100644 index 0000000..cda27c9 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_A_noisy.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_B0.png new file mode 100644 index 0000000..0ddfbaa Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch002_real_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch003_fake_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch003_fake_B0.png new file mode 100644 index 0000000..7aa25d6 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch003_fake_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_A0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_A0.png new file mode 100644 index 0000000..2b0fbb6 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_A0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_A_noisy.png b/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_A_noisy.png new file mode 100644 index 0000000..704bb3f Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_A_noisy.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_B0.png new file mode 100644 index 0000000..0be4875 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch003_real_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch004_fake_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch004_fake_B0.png new file mode 100644 index 0000000..0ed70b3 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch004_fake_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_A0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_A0.png new file mode 100644 index 0000000..6b7b146 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_A0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_A_noisy.png b/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_A_noisy.png new file mode 100644 index 0000000..d80a12b Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_A_noisy.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_B0.png b/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_B0.png new file mode 100644 index 0000000..12f2913 Binary files /dev/null and b/checkpoints/ROMA_UNSB_001/web/images/epoch004_real_B0.png differ diff --git a/checkpoints/ROMA_UNSB_001/web/index.html b/checkpoints/ROMA_UNSB_001/web/index.html index b199ea4..0b79b16 100644 --- a/checkpoints/ROMA_UNSB_001/web/index.html +++ b/checkpoints/ROMA_UNSB_001/web/index.html @@ -4,6 +4,117 @@ Experiment name = ROMA_UNSB_001 +

epoch [4]

+ + + + + + + +
+

+ + +
+

real_A0

+

+ 		+

+ + +
+

real_A_noisy

+

+ 		+

+ + +
+

fake_B0

+

+ 		+

+ + +
+

real_B0

+

+
+

epoch [3]

+ + + + + + + +
+

+ + +
+

real_A0

+

+ 		+

+ + +
+

real_A_noisy

+

+ 		+

+ + +
+

fake_B0

+

+ 		+

+ + +
+

real_B0

+

+
+

epoch [2]

+ + + + + + + +
+

+ + +
+

real_A0

+

+ 		+

+ + +
+

real_A_noisy

+

+ 		+

+ + +
+

fake_B0

+

+ 		+

+ + +
+

real_B0

+

+

epoch [1]

diff --git a/scripts/train.sh b/scripts/train.sh index 77b67e2..9f429ab 100755 --- a/scripts/train.sh +++ b/scripts/train.sh @@ -7,12 +7,12 @@ python train.py \ --dataroot /home/openxs/kunyu/datasets/InfraredCity-Lite/Double/Moitor \ - --name ROMA_UNSB_001 \ + --name ROMA_UNSB_002 \ --dataset_mode unaligned_double \ --no_flip \ --display_env ROMA \ --model roma_unsb \ - --lambda_GAN 8.0 \ + --lambda_GAN 1.0 \ --lambda_NCE 8.0 \ --lambda_SB 1.0 \ --lambda_ctn 1.0 \