ArtiAgent - DefectFill/src/pipeline/__init__.py

"""
Image Artifacts Pipeline

This module provides detection and processing pipeline components for generating
image artifacts using various segmentation models.
"""

# Import GSAMDetector if gsam dependencies exist
try:
    from .gsam_detector import GSAMDetector 
except ImportError:
    GSAMDetector = None 

# Import Flux components if flux dependencies exist
try:
    from .flux_generator import FluxGenerator, FluxConfig
except ImportError:
    FluxGenerator = None
    FluxConfig = None


try:
    from .data_loader import COCODataLoader, ImageNetDataLoader, CustomDirectoryDataLoader
except ImportError:
    COCODataLoader, ImageNetDataLoader, CustomDirectoryDataLoader = None, None, None
    
from .instance_processor import InstanceProcessor


__all__ = [
    'GSAMDetector',
    'FluxGenerator',
    'FluxConfig',
    'COCODataLoader',
    'ImageNetDataLoader',
    'CustomDirectoryDataLoader',
    'InstanceProcessor',
]