Quellcode durchsuchen

update google_utils import

5.0
Glenn Jocher vor 4 Jahren
Ursprung
Commit
6c1b87a42e
4 geänderte Dateien mit 4 neuen und 1 gelöschten Zeilen
  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 Datei anzeigen

@@ -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 Datei anzeigen

@@ -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 Datei anzeigen

@@ -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 Datei anzeigen

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

Laden…
Abbrechen
Speichern