sun15684712291/aicheckv2-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

完成项目推理模块的接口测试

Browse Source
This commit is contained in:
sunyg
2025-04-23 16:38:55 +08:00
parent 0033746fe1
commit 5b38e91f61
8 changed files with 234 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
utils/yolov5/models/yolo.py
View File

@ -25,7 +25,7 @@ if str(ROOT) not in sys.path:
if platform.system() != "Windows":
ROOT = Path(os.path.relpath(ROOT, Path.cwd())) # relative
from models.common import (
from utils.yolov5.models.common import (
C3,
C3SPP,
C3TR,
@ -49,11 +49,11 @@ from models.common import (
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 (
from utils.yolov5.models.experimental import MixConv2d
from utils.yolov5.utils.autoanchor import check_anchor_order
from utils.yolov5.utils.general import LOGGER, check_version, check_yaml, colorstr, make_divisible, print_args
from utils.yolov5.utils.plots import feature_visualization
from utils.yolov5.utils.torch_utils import (
fuse_conv_and_bn,
initialize_weights,
model_info,

4
utils/yolov5/utils/downloads.py
View File

@ -63,7 +63,7 @@ def safe_download(file, url, url2=None, min_bytes=1e0, error_msg=""):
Removes incomplete downloads.
"""
from utils.general import LOGGER
from utils.yolov5.utils.general import LOGGER
file = Path(file)
assert_msg = f"Downloaded file '{file}' does not exist or size is < min_bytes={min_bytes}"
@ -89,7 +89,7 @@ def attempt_download(file, repo="ultralytics/yolov5", release="v7.0"):
"""Downloads a file from GitHub release assets or via direct URL if not found locally, supporting backup
versions.
"""
from utils.general import LOGGER
from utils.yolov5.utils.general import LOGGER
def github_assets(repository, version="latest"):
"""Fetches GitHub repository release tag and asset names using the GitHub API."""