search

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 399 forks source link

Retinanet/MaskRCNN support #280

Open cjpurackal opened 5 years ago

cjpurackal commented 5 years ago

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

dekked commented 5 years ago

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 https://github.com/tryolabs/luminoth/pull/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!

cjpurackal commented 5 years ago

Awesome! will get started with RetinaNet then.