Skip to main content
Associate II
Sep 17, 2024 3:22 PM UTC
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
Sep 20, 2024 8:47 AM UTC

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
Sep 20, 2024 8:47 AM UTC

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.