search

ModelTC / United-Perception

United Perception
Apache License 2.0
428 stars 65 forks source link

Any way to load original yolov5 series model ? #4

Closed Stephenfang51 closed 2 years ago

Stephenfang51 commented 2 years ago

Hi,

Just wonder if we can load original custom pretrained yolov5(s~x) model using your work?

or any way? if so, I suggest updating an easy example.

Thanks your great work !

LitPrice commented 2 years ago

Hi, thanks for your attention. Original yolov5 models can't be loaded directly now. If you want to load, you must rename parameters in yolov5 model. We will consider adding yolov5 head in the future.