updating imports for torchreid

Author: BirenMer

strong_sort/reid_multibackend.py

@@ -9,8 +9,8 @@
 from os.path import exists as file_exists
 from .deep.reid_model_factory import show_downloadeable_models, get_model_url, get_model_name
 
-from torchreid.utils import FeatureExtractor
-from torchreid.utils.tools import download_url
+from torchreid.reid.utils import FeatureExtractor
+from torchreid.reid.utils.tools import download_url
 
 
 def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''):

strong_sort/sort/nn_matching.py

@@ -3,7 +3,7 @@
 import sys
 import torch
 sys.path.append('strong_sort/deep/reid')
-from torchreid.metrics.distance import compute_distance_matrix
+from torchreid.reid.metrics.distance import compute_distance_matrix
 
 
 def _pdist(a, b):

strong_sort/strong_sort.py

@@ -11,8 +11,8 @@
 from .sort.tracker import Tracker
 from .deep.reid_model_factory import show_downloadeable_models, get_model_url, get_model_name
 
-from torchreid.utils import FeatureExtractor
-from torchreid.utils.tools import download_url
+from torchreid.reid.utils import FeatureExtractor
+from torchreid.reid.utils.tools import download_url
 from .reid_multibackend import ReIDDetectMultiBackend
 
 __all__ = ['StrongSORT']