new model

img_demoAE.py

@@ -14,8 +14,8 @@ print(f'loading...')
 
 ########################################################################################################
 
-model_prefix = 'out-v7c_d8_256-224-13bit-OB32x0.5-226'
-input_img = 'kodim19-modified.png'
+model_prefix = 'out-v7c_d8_256-224-13bit-OB32x0.5-298'
+input_imgs = ['lena.png', 'genshin.png', 'kodim14-modified.png', 'kodim19-modified.png', 'kodim24-modified.png']
 device = 'cpu' # cpu cuda
 
 ########################################################################################################
@@ -147,20 +147,22 @@ decoder.load_state_dict(torch.load(f'{model_prefix}-D.pth'))
 
 ########################################################################################################
 
-print(f'test image {input_img}...')
 img_transform = transforms.Compose([
     transforms.PILToTensor(),
     transforms.ConvertImageDtype(torch.float),
     transforms.Resize((224, 224))
 ])
 
-with torch.no_grad():
-    img = img_transform(Image.open(input_img)).unsqueeze(0).to(device)
-    z = encoder(img)
-    z = ToBinary.apply(z)
+for input_img in input_imgs:
+    print(f'test image {input_img}...')
 
-    zz = torch.sum(z.squeeze().long() * zpow, dim=0)
-    print(f'Code shape = {zz.shape}\n{zz.cpu().numpy()}\n')
-    
-    out = decoder(z)
-    vision.utils.save_image(out, f"{input_img.split('.')[0]}-out-13bit.png")
+    with torch.no_grad():
+        img = img_transform(Image.open(f'img_test/{input_img}')).unsqueeze(0).to(device)
+        z = encoder(img)
+        z = ToBinary.apply(z)
+
+        zz = torch.sum(z.squeeze().long() * zpow, dim=0)
+        print(f'Code shape = {zz.shape}\n{zz.cpu().numpy()}\n')
+        
+        out = decoder(z)
+        vision.utils.save_image(out, f"img_test/{input_img.split('.')[0]}-out-13bit.png")

out-v7c_d8_256-224-13bit-OB32x0.5-226-D.pth → out-v7c_d8_256-224-13bit-OB32x0.5-298-D.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2ef336f14ec15ccaf4c1ef25b96b88ca2ee33e105c6275738eabf519e6aae85c
+oid sha256:303a913f6a92286082da49f0b469528008f8585ab5d715114d8d9a2d04fa76ef
 size 25068760

out-v7c_d8_256-224-13bit-OB32x0.5-226-E.pth → out-v7c_d8_256-224-13bit-OB32x0.5-298-E.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ce5f83d8f6b35f84a2db5ca29b5502f22e2ce4af7265d18e476e20a79a366406
+oid sha256:ebdfa80d07808a1ea54176e685dc8321257dd1d574caa361b761cdb795356cab
 size 25076297