Error training object_detection using model-zoo-services

[milanvdm]

Im doing a proof-of-concept on trying to run a model on the STM32N6 that does object-detection.

For this, Im following this guide: https://github.com/STMicroelectronics/stm32ai-modelzoo-services/blob/main/object_detection/src/training/README.md

1. `!git clone https://github.com/STMicroelectronics/stm32ai-modelzoo-services.git`

2. `!pip install -r stm32ai-modelzoo-services/requirements.txt`

3. Im using a Pascal VOC XML dataset which Im converting using the provided scripts with the different config files as attachment in a zip-file

4. `!python converter.py --config-name dataset_config_train.yaml`

5. `!python converter.py --config-name dataset_config_validation.yaml`

6. `!python dataset_create_tfs.py --config-name dataset_config_tfs.yaml`

7. `!python stm32ai_main.py`

This gives the following error (full stacktrace as attachment): `ValueError: Can't convert Python sequence with mixed types to Tensor.`.

As I believe Im following the steps correctly, Im unsure what Im missing to correctly train this model?

[MCHTO.1]

Hello,

It looks like you have a typo after modifying the yaml file.

Can you share your yaml file.

Regards.

[SullyNiu]

@MCHTO.1 

It's the yaml file.

I don’t think it’s a problem with the yaml file. I used the original file on github and got the same error.  You can reproduce this issue according to the above steps(ubuntu conda environment).