mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
update google_utils import
This commit is contained in:
parent
7306dadf78
commit
6c1b87a42e
@ -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 *
|
||||
|
||||
|
1
test.py
1
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 *
|
||||
|
||||
|
1
train.py
1
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 *
|
||||
|
||||
|
@ -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')
|
||||
|
Loading…
x
Reference in New Issue
Block a user