cancel
Showing results for 
Search instead for 
Did you mean: 

pretrained weights with YOLOX

mls
Associate

Hi all,

I'm training a ST_yolox model to detect car. In my user_config.yaml i tried to use this : 
pretrained_weights: coco 

When i launch the training script i got this error

raise ValueError("\nUnknown or unsupported attribute. Received `{}`{}".format(attr, message))

ValueError:

Unknown or unsupported attribute. Received `pretrained_weights`

Please check the 'training.model' section of your configuration file.

But if i understand correctly we can use this variable

https://github.com/STMicroelectronics/stm32ai-modelzoo-services/blob/main/object_detection/docs/README_TRAINING.md#2-7
If someone can give me some explanation 


 

2 REPLIES 2
Julian E.
ST Employee

Hello @mls,

 

You are right, it is in the doc, but the example taken in the doc is the training of a ssd_mobilenetv1

JulianE_0-1762339498103.png

For this type of model, the pretrained weights is available.

 

For yoloX it is not.

I suggest that, for any model, you start from the yaml in the model folder. In your case, this for example:

stm32ai-modelzoo/object_detection/st_yolo_x/ST_pretrainedmodel_public_dataset/coco_2017_80_classes/st_yolo_x_nano_480 at main · STMicroelectronics/stm32ai-modelzoo

 

In the yaml, you can see what seems to be available or not.

 

Please feel free to share any feedback that you may have for us to improve our tools.

 

have a good day,

Julian


In order to give better visibility on the answered topics, please click on 'Accept as Solution' on the reply which solved your issue or answered your question.

Hello @Julian E. 

Thank you for your answer. 

Ah ok ... it's a shame that we can't use pretrained weight on this model ...
Then i have a good dataset. But do you think it's better to use yoloV8 or V11 to detect car ?
In this models (V8 or V11) can i use pretrained weight ?
I don't need fast inference but i need precision ?
Also can i test 480x480 input or 640x640 do you think ?