Blur Detection Final Added

Author: roy-shudipto

__pycache__/blur_detection.cpython-36.pyc

@@ -1,18 +1,39 @@
 import cv2
+import os
 
 
-def variance_of_laplacian(image):
-    return cv2.Laplacian(image, cv2.CV_64F).var()
+def main(blur_thresh=7):
+    folder_list = []
+    if len(os.listdir("Data")) == 0:
+        print("[Data] folder is empty!\n")
+        exit()
+    else:
+        for entry_name in os.listdir("Data"):
+            folder_list.append(entry_name)
+
+    counter = 0
+    for folder in folder_list:
+        folder_path = "Data/" + folder
+        for file_name in os.listdir(folder_path):
+            img = cv2.imread(os.path.join(folder_path, file_name), cv2.IMREAD_COLOR)
 
+            img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+            var = variance_of_laplacian(img_gray)
 
-def main():
-    image = cv2.imread("sharp.jpg")
-    cv2.imshow("Image", image)
-    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    fm = variance_of_laplacian(gray)
+            if var < blur_thresh:
+                counter += 1
+                '''cv2.imshow("Image", img)
+                cv2.waitKey(0)
+                cv2.destroyAllWindows()'''
+                os.remove(os.path.join(folder_path, file_name))
+                print("File Name: " + str(file_name) + " >> Blur detected >> Removed!")
+                print("Original Var: " + str(var))
+                print('--------------------------------')
+    print("Total " + str(counter) + " files removed because of BLUR!")
 
-    print(fm)
-    cv2.waitKey(0)
+
+def variance_of_laplacian(image):
+    return cv2.Laplacian(image, cv2.CV_64F).var()
 
 
 # Main Call Func. >>>

__pycache__/melNET_augData.cpython-36.pyc

@@ -16,7 +16,7 @@ def main():
     global applied
     applied = False
 
-    def apply_button_applied():
+    def apply_but ton_applied():
         root.destroy()
         global applied
         applied = True

__pycache__/melNET_confusion_mat.cpython-36.pyc

@@ -5,6 +5,7 @@
 import os
 from tkinter import *
 import shutil
+import blur_detection
 
 
 # Main >>>
@@ -26,6 +27,7 @@ def quit_button_pressed():
         pressed = False
 
     # Check Box variables
+    _del_blur = BooleanVar()
     _original = BooleanVar()
     _histEqualization = BooleanVar()
     _dilation = BooleanVar()
@@ -36,30 +38,34 @@ def quit_button_pressed():
     _rotate = BooleanVar()
     _five_fold = BooleanVar()
     _rotate_ang = IntVar(0)
+    _blur_thresh = IntVar(0)
 
     # UI Creation
-    Checkbutton(root, text="Original", variable=_original).grid(row=0, sticky=W)
-    Checkbutton(root, text="Hist. Equalization", variable=_histEqualization).grid(row=1, sticky=W)
-    Checkbutton(root, text="Dilation", variable=_dilation).grid(row=2, sticky=W)
-    Checkbutton(root, text="Erosion", variable=_erosion).grid(row=3, sticky=W)
-    Checkbutton(root, text="Blur", variable=_blur).grid(row=4, sticky=W)
-    Checkbutton(root, text="Sharpen", variable=_sharpen).grid(row=5, sticky=W)
-    Checkbutton(root, text="Mirror", variable=_mirror).grid(row=6, sticky=W)
-    Checkbutton(root, text="Rotate All", variable=_rotate).grid(row=7, sticky=W)
-
-    Label(root, text="Angle Span in Degrees: ").grid(row=7, column=2)
-    Label(root, text="(Default: 45)").grid(row=8, column=2)
-    Entry(root, textvariable=_rotate_ang).grid(row=7, column=3)
+    Checkbutton(root, text="Detect & Delete Blurry Images", variable=_del_blur).grid(row=0, sticky=W)
+    Checkbutton(root, text="Original", variable=_original).grid(row=1, sticky=W)
+    Checkbutton(root, text="Hist. Equalization", variable=_histEqualization).grid(row=2, sticky=W)
+    Checkbutton(root, text="Dilation", variable=_dilation).grid(row=3, sticky=W)
+    Checkbutton(root, text="Erosion", variable=_erosion).grid(row=4, sticky=W)
+    Checkbutton(root, text="Blur", variable=_blur).grid(row=5, sticky=W)
+    Checkbutton(root, text="Sharpen", variable=_sharpen).grid(row=6, sticky=W)
+    Checkbutton(root, text="Mirror", variable=_mirror).grid(row=7, sticky=W)
+    Checkbutton(root, text="Rotate All", variable=_rotate).grid(row=8, sticky=W)
+
+    Label(root, text="Angle Span in Degrees: ").grid(row=8, column=2)
+    Label(root, text="(Default: 45)").grid(row=9, column=2)
+    Entry(root, textvariable=_rotate_ang).grid(row=8, column=3)
+    Label(root, text="Blur Threshold (Default=7):").grid(row=0, column=2)
+    Entry(root, textvariable=_blur_thresh).grid(row=0, column=3)
 
     Checkbutton(root, text="Five-Fold   (Default: Single-Fold)", variable=_five_fold).grid(row=10, sticky=W)
     Button(root, text="Quit", command=quit_button_pressed, width=15).grid(row=11, column=2, sticky=W)
     Button(root, text="Apply", command=apply_button_pressed, width=15).grid(row=11, column=3, sticky=W)
 
     root.mainloop()
 
-
     # Checking [Start] status
     if pressed:
+        del_blur = _del_blur.get()
         original = _original.get()
         histEqualization = _histEqualization.get()
         dilation = _dilation.get()
@@ -71,6 +77,7 @@ def quit_button_pressed():
         global five_fold
         five_fold = _five_fold.get()
         rot_ang = _rotate_ang.get()
+        blur_thresh = _blur_thresh.get()
 
         if (original is False and histEqualization is False and dilation is False and
                 erosion is False and blur is False and sharpen is False and mirror is False):
@@ -90,6 +97,12 @@ def quit_button_pressed():
     else:
         rotationAngle = None
 
+    # Detect and Delete blurry images
+    if del_blur:
+        if blur_thresh is 0:
+            blur_thresh = 7
+        blur_detection.main(blur_thresh)
+
     # Number of folds
     global fold_num
     fold_num = 5  # For 5Folds