open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
29.01k stars 9.36k forks source link

graph conv net ontop of Faster RCNN in mmdetection #10996

Open jamesbc123 opened 11 months ago

jamesbc123 commented 11 months ago

Could you suggest where I can implement a graph convolutional network? I would like to take the features from Faster R-CNN and pass them through a graph conv net. I'm unsure if I can do this in a new class (new ROI Head) or if in mm-lab there is some functions available elsewhere for this. I'm also unsure how the computational graph works in mmdetection: if we take the features from Faster R-CNN and pass it through another net, is this other net still able to be updated in back prop?

Thanks for your help

James

jamesbc123 commented 10 months ago

Hello @hhaAndroid, do you have any thoughts on this?

Thank you!