Skip to content

Commit 217b32e

Browse files
authored
Merge pull request #8 from roboflow/fix/optional-imports
fix(trackers): optional import bug for deepsort
2 parents b1a98ad + 8941ca6 commit 217b32e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

trackers/core/deepsort/feature_extractor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import torch
77
import torch.nn as nn
88
import torchvision.transforms as transforms
9-
import validators
109
from firerequests import FireRequests
1110

1211
from trackers.utils.torch_utils import parse_device_spec
@@ -106,6 +105,8 @@ def _initialize_model(
106105
self, model_or_checkpoint_path: Union[str, torch.nn.Module, None]
107106
):
108107
if isinstance(model_or_checkpoint_path, str):
108+
import validators
109+
109110
if validators.url(model_or_checkpoint_path):
110111
checkpoint_path = FireRequests().download(model_or_checkpoint_path)[0]
111112
self._load_model_from_path(checkpoint_path)

0 commit comments

Comments
 (0)