ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
718 stars 209 forks source link

Support for Yolov5 #209

Closed rickymedrano closed 3 years ago

rickymedrano commented 3 years ago

Is there a way I can export the weights and config for yolov5 (.pt and .yaml) to use with tkDNN?

mive93 commented 3 years ago

Hi @rickymedrano, no, we currently do no support Yolov5.