search

leovandriel / caffe2_cpp_tutorial

C++ transcripts of the Caffe2 Python tutorials and other C++ example code
BSD 2-Clause "Simplified" License
431 stars 94 forks source link

Detectron #52

Open dbrazey opened 6 years ago

dbrazey commented 6 years ago

Hello,

Facebook research recently released the Detectron module based on caffe2, to do object detection. This module includes python code. I tried to load pretrained nets with a c++ code inspired from your repo, but it failed.

Did you try to use your code to load such nets ?

Thanks for your work :)

leovandriel commented 6 years ago

Hi, I haven't tried running Detectron in C++. It seems most of the caffe2 detectron module is written in C++, but indeed the main Detectron repo is mostly Python.

Loading pre-trained networks shouldn't be too hard. It's important though that they have identical architecture. So you will need to port the Mask R-CNN model to this repo before you can train it here.