how to use mask-rcnn in caffe2 c++ gpu

pytorch / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

https://pytorch.org

Other

82.54k stars 22.21k forks source link

Open eleven-88 opened 5 years ago

eleven-88 commented 5 years ago

I changed the pkl model to pb model use convert_pkl_to_pb.py , but only detection net will be converted even in Mask R-CNN which has mask net

If anyone could share an example working code with GPU in C++, that'd be amazing.

pascschoSSL commented 5 years ago

@eleven-88 could you solve the problem? I am facing the same problem too, it would be great to have a solution.

LvJC commented 5 years ago

@eleven-88 @pascschoSSL Have you guys got Mask RCNN codes of C++ on caffe2?