File tree Expand file tree Collapse file tree 1 file changed +10
-7
lines changed
stm32ai_application_code/object_detection/Application/STM32H747I-DISCO/Src/CM7 Expand file tree Collapse file tree 1 file changed +10
-7
lines changed Original file line number Diff line number Diff line change @@ -92,13 +92,12 @@ uint8_t lcd_display_global_memory[SDRAM_BANK_SIZE + LCD_FRAME_BUFFER_SIZE];
9292
9393
9494uint8_t pixel_conv_lut [256 ];
95- __attribute__((section (".Out_Postproc" )))
96- #if POSTPROCESS_TYPE == POSTPROCESS_YOLO_V2
97- postprocess_outBuffer_t out_postproc [AI_OBJDETECT_YOLOV2_PP_MAX_BOXES_LIMIT ];
98- #elif POSTPROCESS_TYPE == POSTPROCESS_ST_SSD
95+ #if POSTPROCESS_TYPE == POSTPROCESS_ST_SSD
96+ __attribute__((section (".Out_Postproc" )))
9997 postprocess_outBuffer_t out_postproc [AI_OBJDETECT_SSD_ST_PP_TOTAL_DETECTIONS ];
100- #else
101- postprocess_outBuffer_t out_postproc [];
98+ #elif POSTPROCESS_TYPE == POSTPROCESS_SSD
99+ __attribute__((section (".Out_Postproc" )))
100+ postprocess_outBuffer_t out_postproc [AI_OBJDETECT_SSD_PP_TOTAL_DETECTIONS ];
102101#endif
103102
104103
@@ -243,7 +242,11 @@ static void Software_Init(AppConfig_TypeDef *App_Config_Ptr)
243242 /*Postproc initialization*/
244243 App_Config_Ptr -> error = AI_OBJDETECT_POSTPROCESS_ERROR_NO ;
245244
246- App_Config_Ptr -> output .pOutBuff = & out_postproc [0 ];
245+ #if POSTPROCESS_TYPE == POSTPROCESS_YOLO_V2
246+ App_Config_Ptr -> output .pOutBuff = 0 ;
247+ #else
248+ App_Config_Ptr -> output .pOutBuff = out_postproc ;
249+ #endif
247250 App_Config_Ptr -> error = app_postprocess_init ( App_Config_Ptr );
248251
249252 if (App_Config_Ptr -> error != AI_OBJDETECT_POSTPROCESS_ERROR_NO )
You can’t perform that action at this time.
0 commit comments