Code/Model/src/preprocessing/__init__.py

"""
Preprocessing module for card image processing
"""

from .card_detector import (
    detect_card_boundary,
    detect_card_boundary_strict,
    detect_skin_mask,
    detect_card_boundary_with_hand,
    POKEMON_CARD_DETECTION_CONFIG,
    crop_to_card,
    validate_card_detection,
)
from .image_utils import (
    resize_image,
    normalize_pixels,
    denormalize_pixels,
    check_image_quality,
    adaptive_histogram_equalization,
    remove_noise,
    auto_rotate_card,
    crop_to_content
)
from .augmentation import (
    rotate_image,
    flip_image,
    zoom_image,
    adjust_brightness,
    adjust_contrast,
    adjust_saturation,
    add_gaussian_noise,
    apply_jpeg_compression,
    augment_image,
    augment_dataset,
    validate_augmentation_params
)

__all__ = [
    # Card detection
    "detect_card_boundary",
    "detect_card_boundary_strict",
    "detect_skin_mask",
    "detect_card_boundary_with_hand",
    "POKEMON_CARD_DETECTION_CONFIG",
    "crop_to_card",
    "validate_card_detection",
    # Image utilities
    "resize_image",
    "normalize_pixels",
    "denormalize_pixels",
    "check_image_quality",
    "adaptive_histogram_equalization",
    "remove_noise",
    "auto_rotate_card",
    "crop_to_content",
    # Augmentation - Geometric
    "rotate_image",
    "flip_image",
    "zoom_image",
    # Augmentation - Color
    "adjust_brightness",
    "adjust_contrast",
    "adjust_saturation",
    # Augmentation - Noise
    "add_gaussian_noise",
    "apply_jpeg_compression",
    # Augmentation - Pipeline
    "augment_image",
    "augment_dataset",
    "validate_augmentation_params",
]