docs: def remove_small_boxes(boxlist, min_size):
      """
      Only keep boxes with both sides >= min_size
  
      Arguments:
          boxlist (Boxlist)
          min_size (int)
      """
      # TODO maybe add an API for querying the ws / hs
      xywh_boxes = boxlist.convert("xywh").bbox
      _, _, ws, hs = xywh_boxes.unbind(dim=1)
      keep = (
          (ws >= min_size) & (hs >= min_size)
      ).nonzero().squeeze(1)
      return boxlist[keep]