def get_cells(self): """Return list of all cells (list of Box).""" return [ Box.from_npbox(box_tup) for box_tup in self.cell_to_class_id.keys() ]
def test_from_npbox(self): npbox = np.array([self.ymin, self.xmin, self.ymax, self.xmax]) output_box = Box.from_npbox(npbox) self.assertEqual(output_box, self.box)
def get_boxes(self): """Return list of Boxes.""" return [Box.from_npbox(npbox) for npbox in self.boxlist.get()]