Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
UltralyticsAssistant committed Jun 21, 2024
1 parent 2211ef5 commit 112c6a9
Showing 1 changed file with 29 additions and 3 deletions.
32 changes: 29 additions & 3 deletions models/yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,15 +26,41 @@
ROOT = Path(os.path.relpath(ROOT, Path.cwd())) # relative

from models.common import (
C3, C3SPP, C3TR, SPP, SPPF, Bottleneck, BottleneckCSP, C3Ghost, C3x, Classify, Concat, Contract, Conv,
CrossConv, DetectMultiBackend, DWConv, DWConvTranspose2d, Expand, Focus, GhostBottleneck, GhostConv, Proto,
C3,
C3SPP,
C3TR,
SPP,
SPPF,
Bottleneck,
BottleneckCSP,
C3Ghost,
C3x,
Classify,
Concat,
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.general import LOGGER, check_version, check_yaml, colorstr, make_divisible, print_args
from utils.plots import feature_visualization
from utils.torch_utils import (
fuse_conv_and_bn, initialize_weights, model_info, profile, scale_img, select_device, time_sync,
fuse_conv_and_bn,
initialize_weights,
model_info,
profile,
scale_img,
select_device,
time_sync,
)

try:
Expand Down

0 comments on commit 112c6a9

Please sign in to comment.