C++ Mask R-CNN example

I made C++ implementation of Mask R-CNN with PyTorch C++ frontend. The code is based on PyTorch implementations from multimodallearning and Keras implementation from Matterport . Project was made for educational purposes and can be used as comprehensive example of PyTorch C++ frontend API. Besides regular API you will find how to: load data from MSCoco dataset, create custom layers, manage weights across language boundaries(Python to C++). Feel free to ask a questions.

4 Likes

Can not believe no-one commented on this. It is one of the most comprehensive examples of torchlib.

Thanks.

Hi,
could you please tell me which weights do you use for the inference (demo) and where can one get them?
Can we use the HDF5 file (.h5) as weights or do we need to convert the weights to a format accepted by c++ implementation? More info on this would be appreciated.

Thanks

Hi, please find the instructions how to run demo on the page https://github.com/Kolkir/mlcpp/blob/master/mask_rcnn_pytorch/README.md, there is also a link to parameters.

@kolkir Thank you for the response. I tried that but got some issue with the compilation. Could you provide some inputs on how to convert the model to be read in C++? Was it using β€˜jit’ torchvision?

Please, take a look at this thread How to copy network parameters in libtorch c++ API

@kolkir okay, thank you for link.