ellipse-rcnn icon indicating copy to clipboard operation
ellipse-rcnn copied to clipboard

A PyTorch implementation of Ellipse R-CNN

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.