search

cathy-kim / Pelee-TensorRT

Pelee(NeurIPS'18)-TensorRT Implementation (Caffe Parser)
86 stars 29 forks source link

How can I use my own model? #1

Closed xiaowenhe closed 5 years ago

xiaowenhe commented 5 years ago

Hi, thanks for your work! But how can I use my own model trained with https://github.com/Robert-JunWang/Pelee.git? Thank you !

cathy-kim commented 5 years ago

Hi. You could use your own model by replacing the model and weight file in https://github.com/ginn24/Pelee-TensorRT/tree/master/model/pelee directory.

Thank you.