How to Retrain Classification model?

General
1

Hi

I wonder if there is a guide that summarizes how to retrain the classification model

In the link (hailo_model_zoo/docs/public_models/HAILO8/HAILO8_classification.rst at master · hailo-ai/hailo_model_zoo · GitHub) I found, I found only the tflte model file that was created after it had already been quantized

As far as I know, it’s impossible to retrain these models

Is it possible to export the classification model made through Yolo8n-cls.pt like object detection to onnx and then compile it?

And I also want to know the classification model input image size limit (I understand that object detection is only estimated at 640.)

2

We provide retraining Docker for a few selected models only.

GitHub - Hailo Model Zoo - Retrain on custom dataset

For all other models you will need to consult the original source. The links are available for each model in the Model Zoo. See page you linked above.

If you need some additional help, ChatGPT can give you some really good pointers in many cases.

3

Thank you for your reply

I want to know selected classification model re-train method

Is there no way to re-train the Hailo-supported Classification model?