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];
92
92
93
93
94
94
uint8_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" )))
99
97
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 ];
102
101
#endif
103
102
104
103
@@ -243,7 +242,11 @@ static void Software_Init(AppConfig_TypeDef *App_Config_Ptr)
243
242
/*Postproc initialization*/
244
243
App_Config_Ptr -> error = AI_OBJDETECT_POSTPROCESS_ERROR_NO ;
245
244
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
247
250
App_Config_Ptr -> error = app_postprocess_init ( App_Config_Ptr );
248
251
249
252
if (App_Config_Ptr -> error != AI_OBJDETECT_POSTPROCESS_ERROR_NO )
You can’t perform that action at this time.
0 commit comments