Skip to main content
Associate II
September 17, 2024
Solved

model zoo, dataset

  • September 17, 2024
  • 1 reply
  • 999 views

hello,

I want to train using Model Zoo with a custom dataset. I'm trying to convert the data from COCO format to Darknet format, but I encountered an error with the YAML file (see attached).

Best answer by MCHTO.1

Hello,
It seems that you modified the yaml file structure which caused an issue.
This is an example of correctly structured yaml file that you can use: https://github.com/STMicroelectronics/stm32ai-modelzoo/blob/main/object_detection/datasets/dataset_converter/dataset_config.yaml

Best Regards.


1 reply

MCHTO.1
MCHTO.1Best answer
ST Employee
September 20, 2024

Hello,
It seems that you modified the yaml file structure which caused an issue.
This is an example of correctly structured yaml file that you can use: https://github.com/STMicroelectronics/stm32ai-modelzoo/blob/main/object_detection/datasets/dataset_converter/dataset_config.yaml

Best Regards.