"""
Configuration for ResNet18 AI Image Detector
Hugging Face compatible configuration
"""

from transformers import PretrainedConfig


class ResNet18DetectorConfig(PretrainedConfig):
    """
    Configuration class for ResNet18 AI Image Detector model.
    
    This configuration inherits from PretrainedConfig which contains all the parameters
    to initialize a model with Hugging Face transformers library.
    """
    
    model_type = "resnet18-detector"
    
    def __init__(
        self,
        num_classes: int = 2,
        image_size: int = 224,
        architecture: str = "resnet18",
        dropout_rate: float = 0.5,
        freeze_backbone: bool = False,
        pretrained_weights: str = "IMAGENET1K_V1",
        **kwargs
    ):
        """
        Initialize ResNet18DetectorConfig
        
        Args:
            num_classes (int): Number of output classes (2 for binary classification)
            image_size (int): Input image size (224x224)
            architecture (str): ResNet architecture type
            dropout_rate (float): Dropout rate for regularization
            freeze_backbone (bool): Whether to freeze backbone during training
            pretrained_weights (str): Pretrained weights to use
        """
        self.num_classes = num_classes
        self.image_size = image_size
        self.architecture = architecture
        self.dropout_rate = dropout_rate
        self.freeze_backbone = freeze_backbone
        self.pretrained_weights = pretrained_weights
        
        super().__init__(**kwargs)