custom_ops
custom_ops copied to clipboard
Implement to Matterport/MaskRCNN
Hello @smallsunsun1, I'm trying to implement DCN into Matterport/Mask RCNN but I'm not sure how to do it. I think your code might be able to help me with that but I'm not sure where to start. Can you give me an insight on how to do it?
This repo need to update, DeformablePSROIAlign may have small problem . I have already open a PR in tensorflow_addons, https://github.com/tensorflow/addons/pull/1129, Which provide a keras like layer wrapper for DCN. If you want to start, maybe you can first see the *.cu and *.cpp files, which impelment all the kernels for DCN.