diff --git a/detect.py b/detect.py index 567c9e6..e4a7ddb 100644 --- a/detect.py +++ b/detect.py @@ -2,6 +2,7 @@ import argparse import torch.backends.cudnn as cudnn +from utils import google_utils from utils.datasets import * from utils.utils import * diff --git a/test.py b/test.py index 2381ab4..10de7d8 100644 --- a/test.py +++ b/test.py @@ -3,6 +3,7 @@ import json from torch.utils.data import DataLoader +from utils import google_utils from utils.datasets import * from utils.utils import * diff --git a/train.py b/train.py index 921f0e3..2dfe781 100644 --- a/train.py +++ b/train.py @@ -9,6 +9,7 @@ from torch.utils.tensorboard import SummaryWriter import test # import test.py to get mAP after each epoch from models.yolo import Model +from utils import google_utils from utils.datasets import * from utils.utils import * diff --git a/utils/utils.py b/utils/utils.py index 9dd1d65..cf819c5 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -20,7 +20,7 @@ import yaml from scipy.signal import butter, filtfilt from tqdm import tqdm -from . import torch_utils, google_utils #  torch_utils, google_utils +from . import torch_utils #  torch_utils, google_utils # Set printoptions torch.set_printoptions(linewidth=320, precision=5, profile='long')