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

darknet parser not supported field #222

Closed feritte closed 2 years ago

feritte commented 3 years ago

Hi,

Thanks a lot for your great project.

I exported yolov4-csp model weights trained on a custom dataset by using the given fork of darknet as it is explained on readme. Then using test_yolo4-csp (given the right right names and layers/debug file) I converted it to a rt model. Although it creates a rt model file, i see many "not supported field" lines during the conversion

tkDNN_converting_to_rt_model

tkDnn_converting_2_rt_model_003

At the end I have a rt file which I can load it and use it with demo executable but I would like to know if these are just normal warnings or the conversion is not done correctly.

Thanks again.

Regards

Ferit

mive93 commented 3 years ago

Hi @feritte,

yes, that is totally normal. It's just the parser that do not support those values (that are used for training on darknet, and not useful for inference). If the demo works fine, everything is ok :)