tryolabs / luminoth

Deep Learning toolkit for Computer Vision.

Home Page:https://tryolabs.com

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Retinanet/MaskRCNN support

cjpurackal opened this issue · comments

Hey,
Firstly this is cool project!
I read that you've plans to implement Retinanet and MaskRCNN. If anyone is working on this i'd like to collaborate on it. If not, then I can try implementing these.
I need a green signal from the maintainers to get started.

Thanks,
Christie

Hi @cjpurackal!

Currently, nobody is working on RetinaNet or Mask R-CNN implementation. I guess the simplest to start would be RetinaNet over Faster R-CNN, and take it from there.

Currently we have an open PR #145 with an outdated implementation of RetinaNet that was never merged (it didn't work). We would suggest using the weights of an existing implementation to come up with the forward pass, and only then work on the actual training.

We expect several refactors in the coming months to make Luminoth better, but unfortunately we haven't had much time lately because of client commitments.

Your contribution would be super welcome!

Awesome! will get started with RetinaNet then.