glenn-jocher commited on
Commit
6c1b87a
1 Parent(s): 7306dad

update google_utils import

Browse files
Files changed (4) hide show
  1. detect.py +1 -0
  2. test.py +1 -0
  3. train.py +1 -0
  4. utils/utils.py +1 -1
detect.py CHANGED
@@ -2,6 +2,7 @@ import argparse
2
 
3
  import torch.backends.cudnn as cudnn
4
 
 
5
  from utils.datasets import *
6
  from utils.utils import *
7
 
 
2
 
3
  import torch.backends.cudnn as cudnn
4
 
5
+ from utils import google_utils
6
  from utils.datasets import *
7
  from utils.utils import *
8
 
test.py CHANGED
@@ -3,6 +3,7 @@ import json
3
 
4
  from torch.utils.data import DataLoader
5
 
 
6
  from utils.datasets import *
7
  from utils.utils import *
8
 
 
3
 
4
  from torch.utils.data import DataLoader
5
 
6
+ from utils import google_utils
7
  from utils.datasets import *
8
  from utils.utils import *
9
 
train.py CHANGED
@@ -9,6 +9,7 @@ from torch.utils.tensorboard import SummaryWriter
9
 
10
  import test # import test.py to get mAP after each epoch
11
  from models.yolo import Model
 
12
  from utils.datasets import *
13
  from utils.utils import *
14
 
 
9
 
10
  import test # import test.py to get mAP after each epoch
11
  from models.yolo import Model
12
+ from utils import google_utils
13
  from utils.datasets import *
14
  from utils.utils import *
15
 
utils/utils.py CHANGED
@@ -20,7 +20,7 @@ import yaml
20
  from scipy.signal import butter, filtfilt
21
  from tqdm import tqdm
22
 
23
- from . import torch_utils, google_utils #  torch_utils, google_utils
24
 
25
  # Set printoptions
26
  torch.set_printoptions(linewidth=320, precision=5, profile='long')
 
20
  from scipy.signal import butter, filtfilt
21
  from tqdm import tqdm
22
 
23
+ from . import torch_utils #  torch_utils, google_utils
24
 
25
  # Set printoptions
26
  torch.set_printoptions(linewidth=320, precision=5, profile='long')