Skip to content

Commit d95a31f

Browse files
committed
Fix bug in object detection deployment application.
Signed-off-by: khaoula boutiche <[email protected]>
1 parent fb06762 commit d95a31f

File tree

1 file changed

+10
-7
lines changed
  • stm32ai_application_code/object_detection/Application/STM32H747I-DISCO/Src/CM7

1 file changed

+10
-7
lines changed

stm32ai_application_code/object_detection/Application/STM32H747I-DISCO/Src/CM7/main.c

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -92,13 +92,12 @@ uint8_t lcd_display_global_memory[SDRAM_BANK_SIZE + LCD_FRAME_BUFFER_SIZE];
9292

9393

9494
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")))
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)

0 commit comments

Comments
 (0)