Skip to content

wdoppenberg/ellipse-rcnn

Repository files navigation

PyTorch Lightning

Ellipse R-CNN

A PyTorch (Lightning) implementation of Ellipse R-CNN. Originally developed for another project, it has proven succesful in predicting instanced ellipses. The methodology is based on this paper, albeit different in the sense that this model uses the regressed bounding box predictions instead of region proposals as the base for predicted normalised ellipse parameters.