search

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3
MIT License
487 stars 165 forks source link

Link for caffe models #4

Closed HilmiK closed 5 years ago

HilmiK commented 5 years ago

Hello,

Could you give a link to download sample caffemodels ? And give a clue for training custom models if it is possible using caffe? Or just conversion from darknet ?

lewes6369 commented 5 years ago

Hi, Hilmik.You can check the sample caffemodel here password: es12 It depends on the official 608-yolov3-cfg.

Q:And give a clue for training custom models if it is possible using caffe? Or just conversion from darknet ?

A:Of course, you can train the model in caffe, but you should add the preprocess and postprocess as the sample, use the caffe commit https://github.com/BVLC/caffe/pull/6384/commits/4d2400e7ae692b25f034f02ff8e8cd3621725f5c. Also it could be trained and conversion from darknet .

HilmiK commented 5 years ago

Thanks for reply.