new add

2023-12-22 00:22:44 +08:00 · 2023-12-22 00:22:44 +08:00 · b62dc1b83f
commit b62dc1b83f
parent 9744980ede
3 changed files with 19 additions and 0 deletions
--- a/demo/kodim08.png
+++ b/demo/kodim08.png
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1,6 @@
+torch
+torchvison
+matplotlib
+tqdm
+numpy
+fraction
--- a/scripts.py
+++ b/scripts.py
@ -0,0 +1,13 @@
+import torch
+import torch.nn as nn
+
+
+def image_normalization(norm_type):
+    def _inner(tensor: torch.Tensor):
+        if norm_type == 'nomalization':
+            return tensor / 255.0
+        elif norm_type == 'denormalization':
+            return (tensor * 255.0).type(torch.FloatTensor)
+        else:
+            raise Exception('Unknown type of normalization')
+    return _inner