Unable to generate .nb (NBG) file from custom YOLOv8n model in ST Edge AI Developer Cloud

[OKN]

Hello,

I am trying to create a custom AI model and run inference on the NPU of an STM32MP255F.

At the moment, my goal is only to verify the model creation workflow. Therefore, inference accuracy is not important yet.

Environment

 

• Target device: STM32MP255F

• Inference engine: NPU

• Number of training images: 116 (mouse images)

Steps

1. Dataset creation with Roboflow

• Annotated all images.
• Resized images to 256x256.
• Exported the dataset in YOLOv8 format.

2. Training and export with Ultralytics

Training:

yolo detect train data=data.yaml model=yolov8n.pt epochs=100 imgsz=256

Export:

yolo export model=runs/detect/train/weights/best.pt format=onnx imgsz=256 quantize=8 data=coco8

3. Conversion with ST Edge AI Developer Cloud

• Uploaded the exported model to ST Edge AI Developer Cloud.

• Selected the platform:

  "STM32 Microprocessors embedding Cortex-A loaded with X-LINUX-AI"

• Executed Optimize.

However, the following error is displayed and no .nb file is generated:

Error while generating optimized file. Generation does not contain any output.

 

Additional Information

The Model Zoo object detection model can be converted successfully:

stm32ai-modelzoo/object_detection/yolov8n/
yolov8n_256_quant_pc_uf_od_coco-person-st.tflite

ST Edge AI Developer Cloud generates an .nb file from this model without any issue, and the generated .nb file runs correctly on the STM32MP255F NPU.

Question

Is there anything wrong with my workflow?

Is there any requirement or limitation for ONNX models generated by Ultralytics that would prevent ST Edge AI Developer Cloud from generating an .nb file?

Any advice would be appreciated.

Best Regards,

--

okn

[OKN]

Hello

This is new data created by replacing the original dataset.

Like in the original question, this data also fails the `optimize` step when converting to an .nb file in ST AI Developer Cloud.

 

The attached file is a .zip archive containing best_int8.onnx.

 

Best regards,

--

okn