Source code for birdfsd_yolov5.model_utils.relative_to_abs

#!/usr/bin/env python
# coding: utf-8

from pathlib import Path


def relative_to_abs() -> None:
    """Updates paths in the dataset_config.yaml file from relative to absolute.

    This function is used to replace the dataset path in the
    `dataset_config.yml` file from a relative to absolute path.

    Returns:
        None

    """
    with open('dataset_config.yml') as f:
        lines = f.readlines()

    if '/' not in lines[0]:
        lines[0] = lines[0].replace('dataset-YOLO',
                                    f'{Path().cwd()}/dataset-YOLO')

    with open('dataset_config.yml', 'w') as f:
        f.writelines(lines)
    print(lines)


if __name__ == '__main__':
    relative_to_abs()