image-enhancer-ai 1.0.1__tar.gz

image_enhancer_ai-1.0.1/LICENSE

@@ -0,0 +1,5 @@
+MIT License
+
+Copyright (c) 2026
+
+Permission is hereby granted...

image_enhancer_ai-1.0.1/MANIFEST.in

@@ -0,0 +1,11 @@
+include README.md
+include LICENSE
+recursive-include image_enhancer *.json
+recursive-include image_enhancer *.yaml
+recursive-include image_enhancer *.yml
+recursive-include image_enhancer *.txt
+recursive-include image_enhancer *.h5
+recursive-include image_enhancer *.keras
+recursive-include image_enhancer *.pkl
+recursive-include image_enhancer *.jpg
+recursive-include image_enhancer *.png

image_enhancer_ai-1.0.1/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.4
+Name: image-enhancer-ai
+Version: 1.0.1
+Summary: Reusable AI Image Enhancement Package
+Author-email: Punsara Wikramarathna <punsarawikramarathna@gmail.com>
+License: MIT
+Keywords: image enhancement,opencv,computer vision,image processing,tensorflow,ai
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: opencv-python
+Requires-Dist: numpy
+Requires-Dist: tensorflow
+Requires-Dist: scikit-image
+Requires-Dist: PyYAML
+Dynamic: license-file
+
+# Image Enhancer
+
+A reusable AI-powered image enhancement package.
+
+## Features
+
+- Major Rotation Correction (CNN)
+- Minor Rotation Correction
+- Blur Detection and Correction
+- Noise Reduction
+- Brightness Enhancement
+- Contrast Enhancement
+- Quality Evaluation
+- Iterative Enhancement
+- Configuration Support
+
+## Installation
+
+```bash
+pip install image-enhancer-punsara99
+```
+
+## Usage
+
+```python
+from image_enhancer import ImageEnhancer
+
+engine = ImageEnhancer()
+
+output, report = engine.process("input.jpg")
+
+engine.save(output, "output.jpg")
+
+engine.print_report(report)
+```
+
+## License
+
+MIT

image_enhancer_ai-1.0.1/README.md

@@ -0,0 +1,39 @@
+# Image Enhancer
+
+A reusable AI-powered image enhancement package.
+
+## Features
+
+- Major Rotation Correction (CNN)
+- Minor Rotation Correction
+- Blur Detection and Correction
+- Noise Reduction
+- Brightness Enhancement
+- Contrast Enhancement
+- Quality Evaluation
+- Iterative Enhancement
+- Configuration Support
+
+## Installation
+
+```bash
+pip install image-enhancer-punsara99
+```
+
+## Usage
+
+```python
+from image_enhancer import ImageEnhancer
+
+engine = ImageEnhancer()
+
+output, report = engine.process("input.jpg")
+
+engine.save(output, "output.jpg")
+
+engine.print_report(report)
+```
+
+## License
+
+MIT

image_enhancer_ai-1.0.1/image_enhancer/__init__.py

@@ -0,0 +1 @@
+from .enhancer import ImageEnhancer

image_enhancer_ai-1.0.1/image_enhancer/enhance_engine.py

@@ -0,0 +1,390 @@
+import cv2
+import os
+
+from config.config_loader import load_config
+
+from utils.blur_module import (
+    detect_blur,
+    fix_blur
+)
+
+from utils.noise_module import (
+    remove_noise
+)
+
+from utils.brightness_module import (
+    enhance_brightness_contrast
+)
+
+from utils.quality_module import (
+    quality_score
+)
+
+from utils.rotation_module import (
+    detect_minor_rotation,
+    correct_minor_rotation,
+    rotation_direction,
+    rotation_level
+)
+
+from training.predict_rotation import predict_rotation
+
+from utils.quality_metrics import (
+    improvement,
+    quality_report
+)
+
+
+# ==========================================================
+# Major Rotation Correction
+# ==========================================================
+
+def fix_rotation(image, angle):
+
+    angle = int(float(angle))
+
+    if angle == 90:
+        return cv2.rotate(image, cv2.ROTATE_90_COUNTERCLOCKWISE)
+
+    elif angle == 180:
+        return cv2.rotate(image, cv2.ROTATE_180)
+
+    elif angle == 270:
+        return cv2.rotate(image, cv2.ROTATE_90_CLOCKWISE)
+
+    return image
+
+
+# ==========================================================
+# Configuration Helper
+# ==========================================================
+
+def _enabled(config, section):
+
+    value = config.get(section, {})
+
+    if isinstance(value, dict):
+        return value.get("enabled", False)
+
+    return bool(value)
+
+
+# ==========================================================
+# Run One Enhancement Cycle
+# ==========================================================
+
+def run_pipeline(image, config):
+
+    report = []
+
+    output = image.copy()
+
+    # ---------------------------------------
+    # Blur
+    # ---------------------------------------
+
+    if _enabled(config, "blur"):
+
+        blur_score = detect_blur(output)
+
+        output, _, blur_level, motion, _ = fix_blur(output)
+
+        report.append(f"Blur Score : {blur_score:.2f}")
+        report.append(f"Blur Level : {blur_level}")
+        report.append(f"Motion Blur : {motion}")
+
+    # ---------------------------------------
+    # Noise
+    # ---------------------------------------
+
+    if _enabled(config, "noise"):
+
+        output, noise_score, noise_level, noise_type = remove_noise(output)
+
+        report.append(f"Noise Score : {noise_score:.2f}")
+        report.append(f"Noise Level : {noise_level}")
+        report.append(f"Noise Type : {noise_type}")
+
+    # ---------------------------------------
+    # Brightness
+    # ---------------------------------------
+
+    if _enabled(config, "brightness"):
+
+        output, brightness, b_level, contrast, c_level = \
+            enhance_brightness_contrast(output)
+
+        report.append(f"Brightness : {brightness:.2f}")
+        report.append(f"Brightness Level : {b_level}")
+
+        report.append(f"Contrast : {contrast:.2f}")
+        report.append(f"Contrast Level : {c_level}")
+
+    return output, report
+
+# ==========================================================
+# Image Enhancement Engine
+# ==========================================================
+
+def enhance_image(image):
+
+    if image is None:
+        raise ValueError("Invalid image.")
+
+    config = load_config()
+
+    os.makedirs("output", exist_ok=True)
+
+    original = image.copy()
+    output = image.copy()
+
+    report = []
+
+    # ---------------------------------------------
+    # Track whether image orientation changed
+    # ---------------------------------------------
+
+    rotation_changed = False
+
+    # =====================================================
+    # Major Rotation
+    # =====================================================
+
+    if _enabled(config, "rotation"):
+
+        angle, confidence = predict_rotation(output)
+
+        report.append(f"Major Rotation : {angle}")
+        report.append(f"Confidence : {confidence:.3f}")
+
+        threshold = config["rotation"]["confidence_threshold"]
+
+        if confidence >= threshold and angle != 0:
+
+            output = fix_rotation(output, angle)
+
+            rotation_changed = True
+
+            report.append("Major Rotation Fixed")
+
+        elif angle == 0:
+
+            report.append("No Rotation Needed")
+
+        else:
+
+            report.append("Rotation Ignored (Low Confidence)")
+
+    # =====================================================
+    # Minor Rotation
+    # =====================================================
+
+    if _enabled(config, "minor_rotation"):
+
+        minor_angle = detect_minor_rotation(output)
+
+        direction = rotation_direction(minor_angle)
+
+        level = rotation_level(minor_angle)
+
+        report.append(f"Minor Rotation : {minor_angle:.2f}")
+        report.append(f"Direction : {direction}")
+        report.append(f"Rotation Level : {level}")
+
+        if abs(minor_angle) > 2:
+
+            output = correct_minor_rotation(output, minor_angle)
+
+            rotation_changed = True
+
+            report.append("Minor Rotation Fixed")
+
+    # Save rotation result
+
+    if _enabled(config, "save_steps"):
+
+        cv2.imwrite("output/1_rotation.jpg", output)
+
+    # =====================================================
+    # Iterative Enhancement
+    # =====================================================
+
+    iteration_cfg = config["iteration"]
+
+    max_iterations = iteration_cfg["max_iterations"]
+
+    target_quality = iteration_cfg["quality_threshold"]
+
+    best_quality = -1
+
+    best_output = output.copy()
+
+    best_iteration = 0
+
+    # =====================================================
+    # Enhancement Loop
+    # =====================================================
+
+    if iteration_cfg["enabled"]:
+
+        for i in range(max_iterations):
+
+            report.append("")
+            report.append(f"========== Iteration {i+1} ==========")
+
+            output, temp_report = run_pipeline(output, config)
+
+            report.extend(temp_report)
+
+            current_quality, grade = quality_score(output)
+
+            report.append(f"Current Quality : {current_quality:.2f}")
+
+            # Save intermediate image
+
+            if _enabled(config, "save_steps"):
+
+                cv2.imwrite(
+                    f"output/iteration_{i+1}.jpg",
+                    output
+                )
+
+            # Keep the best result
+
+            # -----------------------------------------
+            # Keep the Best Result
+            # -----------------------------------------
+
+            if current_quality > best_quality:
+
+                best_quality = current_quality
+                best_output = output.copy()
+                best_iteration = i + 1
+
+            else:
+
+                report.append("Quality decreased.")
+                report.append("Stopping iterations.")
+
+                break
+
+            # Stop if target reached
+
+            if current_quality >= target_quality:
+
+                report.append("Target Quality Reached")
+
+                break
+
+    # Use the best enhanced image
+
+    output = best_output.copy()
+
+    final_quality = best_quality
+
+    # =====================================================
+    # Final Quality
+    # =====================================================
+
+    report.append("")
+    report.append("==============================")
+    report.append(" FINAL IMAGE QUALITY ")
+    report.append("==============================")
+
+    report.append(f"Quality Score : {final_quality:.2f}")
+
+    _, grade = quality_score(output)
+
+    report.append(f"Quality Grade : {grade}")
+
+    report.append(f"Iterations Used : {best_iteration}")
+
+    report.append(f"Target Quality : {target_quality}")
+
+    # =====================================================
+    # Quality Metrics
+    # =====================================================
+
+    report.append("")
+    report.append("==============================")
+    report.append(" QUALITY METRICS ")
+    report.append("==============================")
+
+    if rotation_changed:
+
+        report.append("Skipped")
+        report.append(
+            "Reason : Image orientation changed after rotation correction."
+        )
+        report.append(
+            "PSNR/SSIM require identical image orientation."
+        )
+
+    else:
+
+        metrics = quality_report(original, output)
+
+        report.append(f"MSE : {metrics['mse']:.2f}")
+        report.append(f"PSNR : {metrics['psnr']:.2f} dB")
+        report.append(f"SSIM : {metrics['ssim']:.4f}")
+        report.append(f"Image Quality : {metrics['grade']}")
+
+    # =====================================================
+    # Improvement Report
+    # =====================================================
+
+    report.append("")
+    report.append("==============================")
+    report.append(" IMPROVEMENT REPORT ")
+    report.append("==============================")
+
+    before, after, gain = improvement(original, output)
+
+    report.append(f"Before Quality : {before:.2f}")
+    report.append(f"After Quality : {after:.2f}")
+
+    if gain >= 0:
+
+        report.append(f"Improvement : +{gain:.2f}")
+
+    else:
+
+        report.append(f"Improvement : {gain:.2f}")
+
+    # =====================================================
+    # Save Final Image
+    # =====================================================
+
+    if _enabled(config, "save_steps"):
+
+        cv2.imwrite("output/6_final.jpg", output)
+
+    # =====================================================
+    # Completion Summary
+    # =====================================================
+
+    report.append("")
+    report.append("========================================")
+    report.append(" Image Enhancement Completed ")
+    report.append("========================================")
+
+    report.append(f"Quality Score : {final_quality:.2f}")
+    report.append(f"Iterations Used : {best_iteration}")
+    report.append("")
+    report.append("Module Status")
+    report.append("---------------------------")
+
+    report.append("[OK] Rotation")
+    report.append("[OK] Blur")
+    report.append("[OK] Noise")
+    report.append("[OK] Brightness")
+    report.append("[OK] Quality")
+
+    report.append("")
+    report.append("Output Saved Successfully")
+
+    # =====================================================
+    # Return
+    # =====================================================
+
+    return output, report

image_enhancer_ai-1.0.1/image_enhancer/enhancer.py

@@ -0,0 +1,109 @@
+import cv2
+import os
+from config.config_loader import load_config
+from .enhance_engine import enhance_image
+
+
+class ImageEnhancer:
+
+    def __init__(self):
+
+        self.config = load_config()
+
+        print("Image Enhancer Initialized")
+
+    # -----------------------------------------
+    # Show Current Configuration
+    # -----------------------------------------
+
+    def show_configuration(self):
+
+        print("\n========== CURRENT CONFIGURATION ==========\n")
+
+        for key, value in self.config.items():
+
+            print(f"{key} : {value}")
+
+        print()
+
+    # -----------------------------------------
+    # Process Image
+    # -----------------------------------------
+
+    def process(self, image_path):
+
+        image = cv2.imread(image_path)
+
+        if image is None:
+            raise FileNotFoundError(image_path)
+
+        output, report = enhance_image(image)
+
+        return output, report
+
+    # -----------------------------------------
+    # Save Image
+    # -----------------------------------------
+
+    def process_and_save(self, image_path, output_path="output/final.jpg"):
+
+        output, report = self.process(image_path)
+
+        cv2.imwrite(output_path, output)
+
+        return report
+   
+        # -----------------------------------------
+        # Save Image
+        # -----------------------------------------
+
+    def save(self, image, output_path):
+
+            
+
+        folder = os.path.dirname(output_path)
+
+        if folder != "":
+            os.makedirs(folder, exist_ok=True)
+
+            cv2.imwrite(output_path, image)
+
+            print(f"Image saved to {output_path}")
+    # -----------------------------------------
+    # Print Report
+    # -----------------------------------------
+
+    def print_report(self, report):
+
+        print("\n========== REPORT ==========\n")
+
+        for line in report:
+            print(line)
+
+
+    # -----------------------------------------
+    # Batch Processing
+    # -----------------------------------------
+
+    def process_folder(self, input_folder, output_folder="output/results"):
+
+        import os
+
+        os.makedirs(output_folder, exist_ok=True)
+
+        for file in os.listdir(input_folder):
+
+            path = os.path.join(input_folder, file)
+
+            if not os.path.isfile(path):
+                continue
+
+            output, report = self.process(path)
+
+            save_path = os.path.join(output_folder, file)
+
+            cv2.imwrite(save_path, output)
+
+            print(file, "Completed")
+
+        print("\nBatch Processing Finished")

image_enhancer_ai-1.0.1/image_enhancer_ai.egg-info/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.4
+Name: image-enhancer-ai
+Version: 1.0.1
+Summary: Reusable AI Image Enhancement Package
+Author-email: Punsara Wikramarathna <punsarawikramarathna@gmail.com>
+License: MIT
+Keywords: image enhancement,opencv,computer vision,image processing,tensorflow,ai
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: opencv-python
+Requires-Dist: numpy
+Requires-Dist: tensorflow
+Requires-Dist: scikit-image
+Requires-Dist: PyYAML
+Dynamic: license-file
+
+# Image Enhancer
+
+A reusable AI-powered image enhancement package.
+
+## Features
+
+- Major Rotation Correction (CNN)
+- Minor Rotation Correction
+- Blur Detection and Correction
+- Noise Reduction
+- Brightness Enhancement
+- Contrast Enhancement
+- Quality Evaluation
+- Iterative Enhancement
+- Configuration Support
+
+## Installation
+
+```bash
+pip install image-enhancer-punsara99
+```
+
+## Usage
+
+```python
+from image_enhancer import ImageEnhancer
+
+engine = ImageEnhancer()
+
+output, report = engine.process("input.jpg")
+
+engine.save(output, "output.jpg")
+
+engine.print_report(report)
+```
+
+## License
+
+MIT

image_enhancer_ai-1.0.1/image_enhancer_ai.egg-info/SOURCES.txt

@@ -0,0 +1,18 @@
+LICENSE
+MANIFEST.in
+README.md
+pyproject.toml
+image_enhancer/__init__.py
+image_enhancer/config.py
+image_enhancer/enhance_engine.py
+image_enhancer/enhancer.py
+image_enhancer_ai.egg-info/PKG-INFO
+image_enhancer_ai.egg-info/SOURCES.txt
+image_enhancer_ai.egg-info/dependency_links.txt
+image_enhancer_ai.egg-info/requires.txt
+image_enhancer_ai.egg-info/top_level.txt
+tests/test_blur.py
+tests/test_noise.py
+tests/test_quality.py
+tests/test_quality_metrics.py
+tests/test_rotation.py

image_enhancer_ai-1.0.1/image_enhancer_ai.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

image_enhancer_ai-1.0.1/image_enhancer_ai.egg-info/requires.txt

@@ -0,0 +1,5 @@
+opencv-python
+numpy
+tensorflow
+scikit-image
+PyYAML

image_enhancer_ai-1.0.1/image_enhancer_ai.egg-info/top_level.txt

@@ -0,0 +1 @@
+image_enhancer

image_enhancer_ai-1.0.1/pyproject.toml

@@ -0,0 +1,48 @@
+[build-system]
+requires = ["setuptools>=65", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "image-enhancer-ai"
+version = "1.0.1"
+
+description = "Reusable AI Image Enhancement Package"
+
+readme = "README.md"
+
+requires-python = ">=3.9"
+
+license = { text = "MIT" }
+
+authors = [
+    { name = "Punsara Wikramarathna", email = "punsarawikramarathna@gmail.com" }
+]
+
+keywords = [
+    "image enhancement",
+    "opencv",
+    "computer vision",
+    "image processing",
+    "tensorflow",
+    "ai"
+]
+
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent"
+]
+
+dependencies = [
+    "opencv-python",
+    "numpy",
+    "tensorflow",
+    "scikit-image",
+    "PyYAML"
+]
+
+[tool.setuptools]
+include-package-data = true
+
+[tool.setuptools.packages.find]
+include = ["image_enhancer*"]

image_enhancer_ai-1.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

image_enhancer_ai-1.0.1/tests/test_blur.py

@@ -0,0 +1,19 @@
+import cv2
+
+from utils.blur_module import *
+
+img = cv2.imread("input/test.png")
+
+output,score,level,motion,confidence=fix_blur(img)
+
+print("\n========== BLUR REPORT ==========\n")
+
+print("Blur Score :",round(score,2))
+
+print("Blur Level :",level)
+
+print("Motion Blur :",motion)
+
+print("Confidence :",confidence)
+
+cv2.imwrite("../output/test_blur.jpg",output)

image_enhancer_ai-1.0.1/tests/test_quality_metrics.py

@@ -0,0 +1,27 @@
+import cv2
+
+from utils.quality_metrics import *
+
+original = cv2.imread("input/test.png")
+
+enhanced = cv2.imread("output/6_final.jpg")
+
+report = quality_report(original, enhanced)
+
+print()
+
+print("==============================")
+
+print(" IMAGE QUALITY REPORT ")
+
+print("==============================")
+
+print("MSE   :", round(report["mse"],2))
+
+print("PSNR  :", round(report["psnr"],2),"dB")
+
+print("SSIM  :", round(report["ssim"],4))
+
+print("Grade :", report["grade"])
+
+print("==============================")

image_enhancer_ai-1.0.1/tests/test_rotation.py

@@ -0,0 +1,13 @@
+import cv2
+
+from utils.rotation_module import detect_minor_rotation, rotate_image
+
+img=cv2.imread("input/test.png")
+
+angle=detect_minor_rotation(img)
+
+print("Detected Angle :",round(angle,2))
+
+corrected=rotate_image(img,angle)
+
+cv2.imwrite("output/minor_rotation.jpg",corrected)