Skip to content

Conversation

Zeyi-Lin
Copy link
Member

@Zeyi-Lin Zeyi-Lin commented May 20, 2024

Description

环境安装:

pip install ultralytics

测试代码:

from ultralytics import YOLO
from swanlab.integration.ultralytics import add_swanlab_callback


if __name__ == "__main__":
    model = YOLO("yolov8n.yaml")
    model.load()
    add_swanlab_callback(model)

    model.train(
        data="./coco.yaml",
        epochs=50, 
        imgsz=320,
    )

ps: 数据集需要按照Ultralytics文档的指引下载

Closes: #559

@Zeyi-Lin Zeyi-Lin requested a review from SAKURA-CAT May 20, 2024 19:51
@Zeyi-Lin Zeyi-Lin self-assigned this May 20, 2024
@Zeyi-Lin Zeyi-Lin added this to the v0.3.3 milestone May 20, 2024
@Zeyi-Lin Zeyi-Lin added the 💪 enhancement New feature or request label May 21, 2024
@SAKURA-CAT SAKURA-CAT merged commit f9db458 into main May 21, 2024
@SAKURA-CAT SAKURA-CAT deleted the feat/integration-ultralytics branch May 21, 2024 08:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
💪 enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[REQUEST] 集成Ultralytics/YOLO
2 participants