search

lsrock1 / maskrcnn_benchmark.cpp

Implementation maskrcnn-benchmark, pytorch c++ frontend
MIT License
81 stars 23 forks source link

How to train the model to inference? #5

Closed newforrestgump001 closed 5 years ago

newforrestgump001 commented 5 years ago

I wonder how to train the model to inference, By python, which framework ? Thank you a lot! Jack

lsrock1 commented 5 years ago

You can use https://github.com/facebookresearch/maskrcnn-benchmark to train using same config file. After training, you need to convert python pth to cpp pth. My repo has converting script. i will update how to convert.

newforrestgump001 commented 5 years ago

@lsrock1 Thanks for your information! Jack