Droid210/FleetVision

1"""Configuration and constants for damage detector."""2from dataclasses import dataclass3from pathlib import Path4 5EXPECTED_CLASSES = ["Whole", "Damaged"]6NUM_CLASSES = 27MODEL_ID = "microsoft/swin-tiny-patch4-window7-224"8IMAGE_SIZE = 2249 10 11@dataclass12class TrainConfig:13    """Training configuration."""14 15    data_dir: Path16    batch_size: int = 3217    epochs: int = 1518    learning_rate: float = 1e-419    weight_decay: float = 1e-520    num_workers: int = 221    model_output: Path = Path("weights/model b/best_damage_detector.pth")22    seed: int = 4223    recall_weight: float = 2.0  # Weight for recall optimization24