You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I want to deploy a pre-trained object detection model on an STM32 board using STM32Cube.AI.
I deploy the model [ssd_mobilenet_v2_fpnlite_035_192_int8.tflite] pretrained on COCO dataset using the necessary parameters provided in [ssd_mobilenet_v2_fpnlite_035_192_config.yaml] as the example, but I got the following errors:
Hello,
I want to deploy a pre-trained object detection model on an STM32 board using STM32Cube.AI.
I deploy the model [ssd_mobilenet_v2_fpnlite_035_192_int8.tflite] pretrained on COCO dataset using the necessary parameters provided in [ssd_mobilenet_v2_fpnlite_035_192_config.yaml] as the example, but I got the following errors:
I only change the deployment_config.yaml.
deployment_config.yaml.txt
[INFO] : Successfully connected!
[INFO] : Starting the model memory footprints estimation...
Error executing job with overrides: []
Traceback (most recent call last):
File "E:\Desktop\stm32ai-modelzoo-main\object_detection\src\stm32ai_main.py", line 234, in main
process_mode(cfg,
File "E:\Desktop\stm32ai-modelzoo-main\object_detection\src\stm32ai_main.py", line 113, in process_mode
deploy(cfg)
File "E:\Desktop\stm32ai-modelzoo-main\object_detection\src../deployment\deploy.py", line 98, in deploy
stm32ai_deploy(target=board, stlink_serial_number=stlink_serial_number, stm32ai_version=stm32ai_version, c_project_path=c_project_path,
File "E:\Desktop\stm32ai-modelzoo-main\object_detection\src../../common/deployment\common_deploy.py", line 378, in stm32ai_deploy
dispatch_weights(internalFlashSizeFlash_KB=board.config.internalFlash_size,
File "E:\Desktop\stm32ai-modelzoo-main\object_detection\src../../common/deployment\common_deploy.py", line 58, in dispatch_weights
sorted_weights = dict(sorted(graph["weights"].items(), key=lambda item: item[1]['buffer_c_count'], reverse=True))
File "E:\Desktop\stm32ai-modelzoo-main\object_detection\src../../common/deployment\common_deploy.py", line 58, in
sorted_weights = dict(sorted(graph["weights"].items(), key=lambda item: item[1]['buffer_c_count'], reverse=True))
KeyError: 'buffer_c_count'
Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
How can I sovle the problem?
Thank you for your guidance.
The text was updated successfully, but these errors were encountered: