2023-02-28 07:21 AM
Hello,
I'm sorry that these are in fact many questions but I'm wondering why I can't analyse any of the ONNX models that I've downloaded. I tryied YOLOv8n (12MB), YOLOv6n (18MB), YOLOv5su (36MB), YOLOv4tiny (24MB) and YOLOv2 (62 MB) and none of these ONNX models can pass the analysis step except YOLOv2 (but it is oviously too big for the MCU).
I have two main questions.
I'm attaching a zip containing all ONNX files and the associated txt files with the errors.
Thanks a lot,
Camille RAHI
Data Scientist @ ALTEN
2023-02-28 08:45 AM
Several of the H7 platforms can have external OSPI/QSPI NOR flash in the 64 or 128MB (256MB Max)
Similar amounts of SDRAM can also be available externally.
I'm not sure Vincent has a presence here, would suggest you contact directly, or get a referral to the users in question.
https://www.linkedin.com/in/vincent-richard-0bb17217/?originalSubdomain=fr
2023-02-28 08:47 AM
Yes you're right the Yolo model is the one from the model zoo at https://github.com/STMicroelectronics/stm32ai-modelzoo, please contact Vincent if you want to reuse it with another dataset.
If you are looking for a full example of object detection running on stm32 you can also use the ssd_mobilenetv1 models https://github.com/STMicroelectronics/stm32ai-modelzoo/blob/main/object_detection/models/ssd_mobilenetv1/README.md
As a side note, recent versions of Yolo are with a GPL license that potentially contaminate the rest of your application (you'll have to distribute also your application under GPL)
2023-02-28 08:59 AM
YoloV5 (and I guess above) contains a 5D initializer that is not currently supported in stm32ai
Regards
Daniel
2023-03-10 01:27 AM
Hi,
Thanks a lot for your quick answers, that has been really helpful for the past weeks, it's clearer now.
For now we are going to fine-tune your mobilenet model until we get an MP1 board for bigger models.
Any planned release for improved support of ONNX models?
Thanks,
Camille
2024-04-30 08:25 AM
Is it possible to add the custom post-processing function in the middleware of the object detection application https://github.com/STMicroelectronics/stm32ai-modelzoo/tree/main/stm32ai_application_code/object_detection?