Sahi and Yolov8 compatibilitie issue: Yolov5 data export as .yaml or .yml?

[jokober]

Hi, I converted coco data to yolov5 data and then tried to train yolov8 on it. However I get the error that "/data.yaml' does not exist". Looking at the stacktrace below I realised yolov8 is looking for a ".yaml" file and sahi is exporting as .yml file.

        # Model and Dataset
        self.model = self.args.model
        try:
            if self.args.task == 'classify':
                self.data = check_cls_dataset(self.args.data)
            elif self.args.data.endswith('.yaml') or self.args.task in ('detect', 'segment'):
                self.data = check_det_dataset(self.args.data)
                if 'yaml_file' in self.data:
                    self.args.data = self.data['yaml_file']  # for validating 'yolo train data=url.zip' usage
        except Exception as e:
>           raise RuntimeError(emojis(f"Dataset '{clean_url(self.args.data)}' error ❌ {e}")) from e
E           RuntimeError: Dataset '/tmp/pytest-of-jo/pytest-61/local_data0/datasets/200_300_0_1_0_1_0_2/yolov5_files/data.yaml' error ❌ '/tmp/pytest-of-jo/pytest-61/local_data0/datasets/200_300_0_1_0_1_0_2/yolov5_files/data.yaml' does not exist

/home/jo/.cache/pypoetry/virtualenvs/birdmot-s5Dc3Lay-py3.9/lib/python3.9/site-packages/ultralytics/yolo/engine/trainer.py:126: RuntimeError

Here is the line of code in sahi which exports as .yml file.

sahi/sahi/utils/coco.py

Line 1381 in 26acb89

yaml_path = str(Path(output_dir) / "data.yml")

Not sure if you are aware of this but probably you want to change the line of code in order to export as .yaml file?