porting yolov11 model to stm32mp257.(Segmentation fault (core dumped))

[fanronghua0123456]

this is issue when i porting yolov11 to stm32mp257.

 # Reading input image
 input_width = input_tensor_shape[1]
 print(input_width)
 input_height = input_tensor_shape[2]
 print(input_height)
 input_image = Image.open(args.image).resize((input_width,input_height))
 input_data = np.expand_dims(input_image, axis=0)
 if input_tensor_dtype == np.float32:
 input_data = (np.float32(input_data) - args.input_mean) /args.input_std
 print("----1")
 img_array_after = np.array(input_data)
 print("Dtype after resize: ", img_array_after.dtype)
 print("Shape after resize: ", img_array_after.shape)
 print("----1test")
 if input_tensor_dtype == np.float32:
 print("float32")
 if input_tensor_dtype == np.float16:
 print("float16")
 input_data = (np.float16(input_data) - args.input_mean) /args.input_std
 print("----2test")
 stai_model.set_input(0, input_data)
 print("----2")
 start = timer()
 stai_model.run()
 end = timer()

attach is my python files and model.

root@ATK-DLMP257:/opt/ui/src/apps/resource/x-linux-ai/object-detection# python3 testpy.py -m /home/root/best_integer_quant.nb -i /home/root/hander1_1.jpg -l /home
/root/best.txt
Loading dynamically: /usr/lib/libstai_mpu_ovx.so.6
[OVX]: Loading nbg model
**Input node: 0 -Input_name: -Input_dims:4 - input_type:float16 -Input_shape:(1, 640, 640, 3)
**Output node: 0 -Output_name: -Output_dims:3 - Output_type:float16 -Output_shape:(1, 11, 8400)
640
640
----1
Dtype after resize: uint8
Shape after resize: (1, 640, 640, 3)
----1test
float16
----2test
Segmentation fault (core dumped)
root@ATK-DLMP257:/opt/ui/src/apps/resource/x-linux-ai/object-detection#

 

thanks

[Julian E.]

Hi @fanronghua0123456,

 

I think this question is part of this thread. Please don't duplicate threads.

Re: porting yolov11 model to stm32mp257. - Page 2 - STMicroelectronics Community

 

Have a good day,
Julian