cancel
Showing results for 
Search instead for 
Did you mean: 

Training a model

FabioT2801
Associate

Hello, I am trying to train your models on a custom dataset created with​ the STM32F401 and X-NUCLEO 53L5A1 expansion board, and I am new to these things. I have downloaded the hand_posture folder on the model zoo section on GitHub, and followed all the instructions to modify the yaml file and preparing the dataset. But when I run the stm32ai_main.py file it tells me that not all dependencies have been iinstalled, even for other .py files that run with that main file. After I tried to find myself some of these missing dependencies (like Keras, Hydra and other stuff) there are many more missing and I could not find any of these online. Is there a repository to find the necessary files to run the training without errors?

1 REPLY 1
Julian E.
ST Employee

Hello @FabioT2801,

 

In the cloned repository, you should have a file named requierements.txt

You need to run the following command in a cmd to install all the dependencies: pip install -r requierements.txt

Make sure to be in the folder where the requierement file is.

 

For more details, please check the before you start section here: GitHub - STMicroelectronics/stm32ai-modelzoo: AI Model Zoo for STM32 devices

 

Have a good day.

Julian


In order to give better visibility on the answered topics, please click on 'Accept as Solution' on the reply which solved your issue or answered your question.