Selective Search reimplementation in PyTorch that allows to extract not only the bounding boxes, but also the region masks