siammot/modelling/box_head/inference.py (1 line): - line 62: # TODO think about a representation of batch of boxes