123456/roma_unsb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

running without cnt named 001

Browse Source
This commit is contained in:
bishe 2025-02-24 23:10:23 +08:00
parent 4af0d7463d
commit c2e6cfe0b1
8 changed files with 59 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
checkpoints/ROMA_UNSB_001/loss_log.txt
View File

@ -69,3 +69,15 @@
================ Training Loss (Sun Feb 23 23:13:59 2025) ================
================ Training Loss (Sun Feb 23 23:14:59 2025) ================
================ Training Loss (Mon Feb 24 22:59:41 2025) ================
================ Training Loss (Mon Feb 24 23:01:03 2025) ================
================ Training Loss (Mon Feb 24 23:02:59 2025) ================
================ Training Loss (Mon Feb 24 23:07:07 2025) ================
(epoch: 1, iters: 100, time: 0.155, data: 0.229) G_GAN: 6.946 D_real_ViT: 0.807 D_fake_ViT: 0.309 G: 7.258 SB: 0.056
(epoch: 1, iters: 200, time: 0.174, data: 0.004) G_GAN: 5.494 D_real_ViT: 0.538 D_fake_ViT: 0.287 G: 5.808 SB: 0.054
(epoch: 1, iters: 300, time: 0.185, data: 0.003) G_GAN: 4.615 D_real_ViT: 0.424 D_fake_ViT: 0.302 G: 4.865 SB: 0.073
(epoch: 1, iters: 400, time: 0.191, data: 0.003) G_GAN: 4.170 D_real_ViT: 0.357 D_fake_ViT: 0.315 G: 4.751 SB: 0.055
(epoch: 1, iters: 500, time: 0.196, data: 0.004) G_GAN: 4.437 D_real_ViT: 0.323 D_fake_ViT: 0.265 G: 4.986 SB: 0.062
(epoch: 1, iters: 600, time: 0.199, data: 0.003) G_GAN: 4.024 D_real_ViT: 0.289 D_fake_ViT: 0.252 G: 4.247 SB: 0.073
(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

BIN
checkpoints/ROMA_UNSB_001/web/images/epoch001_fake_B0.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 25 KiB

BIN
checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A0.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 38 KiB

BIN
checkpoints/ROMA_UNSB_001/web/images/epoch001_real_A_noisy.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 128 KiB

BIN
checkpoints/ROMA_UNSB_001/web/images/epoch001_real_B0.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 115 KiB

45
checkpoints/ROMA_UNSB_001/web/index.html Normal file
View File

@ -0,0 +1,45 @@
<!DOCTYPE html>
<html>
<head>
<title>Experiment name = ROMA_UNSB_001</title>
</head>
<body>
<h3>epoch [1]</h3>
<table border="1" style="table-layout: fixed;">
<tr>
<td halign="center" style="word-wrap: break-word;" valign="top">
<p>
<a href="images/epoch001_real_A0.png">
<img src="images/epoch001_real_A0.png" style="width:256px">
</a><br>
<p>real_A0</p>
</p>
</td>
<td halign="center" style="word-wrap: break-word;" valign="top">
<p>
<a href="images/epoch001_real_A_noisy.png">
<img src="images/epoch001_real_A_noisy.png" style="width:256px">
</a><br>
<p>real_A_noisy</p>
</p>
</td>
<td halign="center" style="word-wrap: break-word;" valign="top">
<p>
<a href="images/epoch001_fake_B0.png">
<img src="images/epoch001_fake_B0.png" style="width:256px">
</a><br>
<p>fake_B0</p>
</p>
</td>
<td halign="center" style="word-wrap: break-word;" valign="top">
<p>
<a href="images/epoch001_real_B0.png">
<img src="images/epoch001_real_B0.png" style="width:256px">
</a><br>
<p>real_B0</p>
</p>
</td>
</tr>
</table>
</body>
</html>

BIN
models/__pycache__/roma_unsb_model.cpython-39.pyc
View File

Binary file not shown.

5
models/roma_unsb_model.py
View File

@ -239,9 +239,8 @@ class RomaUnsbModel(BaseModel):
BaseModel.__init__(self, opt)
# 指定需要打印的训练损失
self.loss_names = ['G_GAN_1', 'D_real_1', 'D_fake_1', 'G_1', 'NCE_1', 'SB_1',
'G_2']
self.visual_names = ['real_A', 'real_A_noisy', 'fake_B', 'real_B']
self.loss_names = ['G_GAN', 'D_real_ViT', 'D_fake_ViT', 'G', 'SB']
self.visual_names = ['real_A0', 'real_A_noisy', 'fake_B0', 'real_B0']
self.atten_layers = [int(i) for i in self.opt.atten_layers.split(',')]
if self.opt.phase == 'test':