Pārlūkot izejas kodu

update google_utils import

5.0
Glenn Jocher pirms 4 gadiem
vecāks
revīzija
6c1b87a42e
4 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  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 Parādīt failu

@@ -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 Parādīt failu

@@ -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 Parādīt failu

@@ -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 Parādīt failu

@@ -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')

Notiek ielāde…
Atcelt
Saglabāt