from pathlib import Path

import argparse

from thyroid_vfm.config import DetectionConfig, load_yaml_config
from thyroid_vfm.training.detection import train_detection


def main() -> None:
    parser = argparse.ArgumentParser(description="Train YOLO on TN5000 detection.")
    parser.add_argument(
        "--config",
        type=Path,
        default=Path("configs/detection_yolov8.yaml"),
        help="Path to the detection config file.",
    )
    args = parser.parse_args()
    config = load_yaml_config(args.config, DetectionConfig)
    train_detection(config)


if __name__ == "__main__":
    main()