Spaces:

yetrun
/

general-deep-learning

Sleeping

general-deep-learning / src /deep_learning /models /yolo /__init__.py

ver3: 将源码迁入 src/deep_learning 包，重塑训练流水线，规范 data/model 契约

07cb7d3 3 months ago

297 Bytes

from .losses import box_loss
from .inference_tools import draw_prediction, resolve_label_name
from .model_builder import YoloModelBuilder, build_yolo_preprocessor

__all__ = [
    "build_yolo_preprocessor",
    "YoloModelBuilder",
    "box_loss",
    "draw_prediction",
    "resolve_label_name"
]

	from .losses import box_loss
	from .inference_tools import draw_prediction, resolve_label_name
	from .model_builder import YoloModelBuilder, build_yolo_preprocessor

	__all__ = [
	"build_yolo_preprocessor",
	"YoloModelBuilder",
	"box_loss",
	"draw_prediction",
	"resolve_label_name"
	]