add fls

LICENSE

@@ -0,0 +1,13 @@
+DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
+                   Version 2, December 2004
+
+Copyright (C) 2004 Sam Hocevar <[email protected]>
+
+Everyone is permitted to copy and distribute verbatim or modified
+copies of this license document, and changing it is allowed as long
+as the name is changed.
+
+           DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
+  TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
+
+ 0. You just DO WHAT THE FUCK YOU WANT TO.

count_attack.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+
+# https://www.cyberforum.ru/python-beginners/thread2510792.html
+ 
+with open("attack.txt", "rt") as file:
+    text = file.read()
+ 
+for digit in "6789":
+    print(f"{digit} {text.count(digit)}")
+
+# 6 354
+# 7 383
+# 8 359
+# 9 344

resize-and-rename-data.py

@@ -0,0 +1,76 @@
+import PIL
+from PIL import Image
+import os, sys
+import time
+
+
+Width, Height = 480, 320
+
+path1 = "DATASETs/wm5/" 
+color_mode1 = "L"
+
+path2 = "DATASETs/cover5/" 
+color_mode2 = "RGB" 
+
+
+def resize(path, color_mode):
+    dirs = os.listdir(path)
+    print('before resize ', len(dirs))
+    for item in dirs:
+        try:
+            # print(item)
+            with Image.open(fr'{path}{item}') as im:
+                resized = im.convert(f'{color_mode}').resize((Width,Height))
+                resized.save(fr'{path}{item}')
+                time.sleep(0.0003)
+                # print(fr'for {item} have been done')
+        except PIL.UnidentifiedImageError:
+            print(fr"Confirmed: This image {path}{item} cannot be opened!")
+            # os.remove(f'{path}{item}')
+        except OSError:
+            im = Image.open(fr'{path}{item}').convert(f'{color_mode}').resize((Width,Height))
+            im.save(fr'{path}{item}')
+            print(fr"Chanched by hands for {path}{item}")
+    dirs = os.listdir(path)
+    print('after resize ', len(dirs))
+
+
+resize(path1, color_mode1)
+resize(path2, color_mode2)
+
+
+def test_size(path):
+    dirs = os.listdir(path)
+    print('before test ', len(dirs))
+    for item in dirs:
+        try:
+            with Image.open(fr'{path}{item}') as im:
+                width, height = im.size
+                if (width == Width) and (height == Height):
+                    pass
+                else:
+                    print(fr'for {item} not true size')
+                time.sleep(0.0003)
+        except PIL.UnidentifiedImageError:
+            print(fr"Confirmed: This image {item} cannot be opened! We removed it")
+            os.remove(f'{path}{item}')
+    dirs = os.listdir(path)
+    print('after test ', len(dirs))
+
+
+test_size(path1)
+test_size(path2)
+
+
+def renameimg(path):
+    os.getcwd()
+    # print(os.getcwd())
+    for i, filename in enumerate(os.listdir(path)):
+        try:
+            os.rename(path + "/" + filename, path + "/" + str(i) + ".jpeg")
+        
+        except FileExistsError:
+            pass
+
+renameimg(path1)
+renameimg(path2)