How to use yolov8 on STMh7

[dogg]

Hi,

I have successfully ran and trained the models in modelzoo but I would like to use a larger model with yolov8 for higher resolution.

It was mentioned in another post that yolov5 and above is not compatible with ST chips at the moment.

In any case can how can I use a previous model let's say v4? The available models seem to be:

st_ssd_mobilenet_v1

ssd_mobilenet_v2_fpnlite

st_yolo_lc_v1

tiny_yolo_v2

What would the .yaml file look like. Can I just use the .tflite file from yolov4 and use one of the yolo derivatives from the above list?

Alternatively, can I increase the input matrix on the available models to be larger than 416x416?

I am using the h747 disco board.

thanks

[Julian E.]

Hello @dogg,

For the moment, as you said, only the following models are available for object detection:

st_ssd_mobilenet_v1

ssd_mobilenet_v2_fpnlite

st_yolo_lc_v1

tiny_yolo_v2

To deploy it, you can look at this configuration example:

yaml: https://github.com/STMicroelectronics/stm32ai-modelzoo/blob/main/object_detection/src/config_file_examples/deployment_config.yaml 
Doc: https://github.com/STMicroelectronics/stm32ai-modelzoo/tree/main/object_detection/deployment 

Simply put the path to the .tflite model you want to deploy.

Regarding input size, you can change it, but you will need to retrain the model.

(for tiny yolov2 type the input size should be a multiple of 32, and for st__yolo_lc_v1 the input size should be  a  multiple of 16.)

In the yaml, in the training part, you have a input_size field.

DOC training: https://github.com/STMicroelectronics/stm32ai-modelzoo/blob/main/object_detection/src/training/README.md 

Have a good day,

Julian