diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py index ce25c133600a4eeef88309594f11acc87bc7fa37..b1e8a356edecaf980aa687b4fad15115c87f7462 100644 --- a/dan/predict/prediction.py +++ b/dan/predict/prediction.py @@ -2,6 +2,7 @@ import os import pickle +from itertools import pairwise from pathlib import Path import cv2 @@ -20,7 +21,7 @@ from dan.predict.attention import ( plot_attention, split_text_and_confidences, ) -from dan.utils import pairwise, read_image +from dan.utils import read_image class DAN: diff --git a/dan/utils.py b/dan/utils.py index 7325b2b379121f55ecf738c8e78b43f647fda067..93243fcfd8f46876b74027e897c1faf31d5ff5b0 100644 --- a/dan/utils.py +++ b/dan/utils.py @@ -1,6 +1,4 @@ # -*- coding: utf-8 -*- -from itertools import tee - import cv2 import numpy as np import torch @@ -154,13 +152,3 @@ def round_floats(float_list, decimals=2): Round list of floats with fixed decimals """ return [np.around(num, decimals) for num in float_list] - - -def pairwise(iterable): - """ - Not necessary when using 3.10. See https://docs.python.org/3/library/itertools.html#itertools.pairwise. - """ - # pairwise('ABCDEFG') --> AB BC CD DE EF FG - a, b = tee(iterable) - next(b, None) - return zip(a, b)