Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Error in development Object Detection using the given example #52

Open
Hood7319 opened this issue Nov 11, 2024 · 0 comments
Open

Error in development Object Detection using the given example #52

Hood7319 opened this issue Nov 11, 2024 · 0 comments
Assignees

Comments

@Hood7319
Copy link

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants