Skip to content

boundingboxdataiterator

boundingboxdataiterator

Module containing BoundingBoxIterator

Classes

BoundingBoxIterator 

BoundingBoxIterator(parq_extension='.parq')

Bases: TensordataIterator

Iterator for iterating over prediction data

Source code in niceml/data/dataiterators/boundingboxdataiterator.py 
35
36
37
38
39
def __init__(self, parq_extension: str = ".parq"):
    self.parq_extension = parq_extension
    self.target_cols_prefix = "pred"
    self.data = None
    self.keys = None

ObjDetPredictionContainer dataclass

Container class for iterating over prediction data

Functions