Skip to main content
Associate II
September 17, 2024
Solved

model zoo, dataset

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).

This topic has been closed for replies.
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.


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.