Переглянути джерело

update google_utils import

5.0
Glenn Jocher 4 роки тому
джерело
коміт
6c1b87a42e
4 змінених файлів з 4 додано та 1 видалено
  1. +1
    -0
      detect.py
  2. +1
    -0
      test.py
  3. +1
    -0
      train.py
  4. +1
    -1
      utils/utils.py

+ 1
- 0
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 *


+ 1
- 0
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
- 0
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 *


+ 1
- 1
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')

Завантаження…
Відмінити
Зберегти