Implement to Matterport/MaskRCNN

smallsunsun1 / custom_ops

Deformable Convolution Networks(deformable conv2D TF_implementation)

27 stars 8 forks source link

Implement to Matterport/MaskRCNN #2

Open keekw opened 4 years ago

keekw commented 4 years ago

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?

smallsunsun1 commented 4 years ago

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.