HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.3.59 Add ability to load any torchvision model as module (#18564)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Mohammed Yasin 2025-01-09 20:57:46 +08:00 • committed by GitHub
parent 246c3eca81
commit cc1e77138c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 104 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/nn/modules/__init__.py
View file

@ -56,6 +56,7 @@ from .block import (
RepVGGDW,
ResNetLayer,
SCDown,
TorchVision,
)
from .conv import (
CBAM,
@ -68,6 +69,7 @@ from .conv import (
DWConvTranspose2d,
Focus,
GhostConv,
Index,
LightConv,
RepConv,
SpatialAttention,
@ -156,4 +158,6 @@ __all__ = (
"C2fCIB",
"Attention",
"PSA",
"TorchVision",
"Index",
)