"""
Aegis Vision - Computer Vision Training Utilities

A Python package for training computer vision models on cloud platforms like Kaggle.
Focused on YOLO object detection with support for multiple model variants and export formats.
"""

__version__ = "0.2.4"
__author__ = "Aegis AI Team"
__license__ = "MIT"

from .trainer import YOLOTrainer
from .converters import COCOConverter, DatasetMerger, AdvancedCOCOtoYOLOMerger
from .dataset_utils import discover_datasets, preprocess_datasets, preprocess_coco_standard
from .utils import (
    setup_logging,
    get_device_info,
    detect_environment,
    format_size,
    format_time,
)
from .kaggle_uploader import KaggleModelUploader, upload_trained_model

__all__ = [
    "YOLOTrainer",
    "COCOConverter",
    "DatasetMerger",
    "AdvancedCOCOtoYOLOMerger",
    "discover_datasets",
    "preprocess_datasets",
    "preprocess_coco_standard",
    "setup_logging",
    "get_device_info",
    "detect_environment",
    "format_size",
    "format_time",
    "KaggleModelUploader",
    "upload_trained_model",
]
