Use named imports in models.yolo (#12576)
This commit is contained in:
1 parent
46ae996cb1
commit
f6a781ac76
1 file changed
+9
-3
+9
-3
@@ -8,12 +8,16 @@ Usage:
|
|||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
import contextlib
|
import contextlib
|
||||||
|
import math
|
||||||
import os
|
import os
|
||||||
import platform
|
import platform
|
||||||
import sys
|
import sys
|
||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
import torch
|
||||||
|
import torch.nn as nn
|
||||||
|
|
||||||
FILE = Path(__file__).resolve()
|
FILE = Path(__file__).resolve()
|
||||||
ROOT = FILE.parents[1] # YOLOv5 root directory
|
ROOT = FILE.parents[1] # YOLOv5 root directory
|
||||||
if str(ROOT) not in sys.path:
|
if str(ROOT) not in sys.path:
|
||||||
@@ -21,10 +25,12 @@ if str(ROOT) not in sys.path:
|
|||||||
if platform.system() != 'Windows':
|
if platform.system() != 'Windows':
|
||||||
ROOT = Path(os.path.relpath(ROOT, Path.cwd())) # relative
|
ROOT = Path(os.path.relpath(ROOT, Path.cwd())) # relative
|
||||||
|
|
||||||
from models.common import * # noqa
|
from models.common import (C3, C3SPP, C3TR, SPP, SPPF, Bottleneck, BottleneckCSP, C3Ghost, C3x, Classify, Concat,
|
||||||
from models.experimental import * # noqa
|
Contract, Conv, CrossConv, DetectMultiBackend, DWConv, DWConvTranspose2d, Expand, Focus,
|
||||||
|
GhostBottleneck, GhostConv, Proto)
|
||||||
|
from models.experimental import MixConv2d
|
||||||
from utils.autoanchor import check_anchor_order
|
from utils.autoanchor import check_anchor_order
|
||||||
from utils.general import LOGGER, check_version, check_yaml, make_divisible, print_args
|
from utils.general import LOGGER, check_version, check_yaml, colorstr, make_divisible, print_args
|
||||||
from utils.plots import feature_visualization
|
from utils.plots import feature_visualization
|
||||||
from utils.torch_utils import (fuse_conv_and_bn, initialize_weights, model_info, profile, scale_img, select_device,
|
from utils.torch_utils import (fuse_conv_and_bn, initialize_weights, model_info, profile, scale_img, select_device,
|
||||||
time_sync)
|
time_sync)
|
||||||
|
|||||||
Reference in new issue
Block a user