Skip to content

feat: MindNLP兼容Ultralytics库#2762

Open
kittentruck wants to merge 6 commits into
mindspore-lab:masterfrom
kittentruck:feat-ultralytics
Open

feat: MindNLP兼容Ultralytics库#2762
kittentruck wants to merge 6 commits into
mindspore-lab:masterfrom
kittentruck:feat-ultralytics

Conversation

@kittentruck
Copy link
Copy Markdown

@kittentruck kittentruck commented Mar 9, 2026

完成 MindSpore NLP 对 Ultralytics 库的兼容适配(#1805
链接为https://github.com/kittentruck/mindnlp/pull/new/feat-ultralytics

@moyu026
Copy link
Copy Markdown
Contributor

moyu026 commented Mar 19, 2026

./ultralytics/cfg/datasets,这个路径下没有datasets

@kittentruck
Copy link
Copy Markdown
Author

已添加缺失的数据配置文件 ,.yaml 文件中的 path、train 和 val 路径可能需要修改一下

@moyu026
Copy link
Copy Markdown
Contributor

moyu026 commented Mar 19, 2026

写下运行需要的requirements.txt

@kittentruck
Copy link
Copy Markdown
Author

已添加requirements.txt文件,并修改了其它文件

@moyu026
Copy link
Copy Markdown
Contributor

moyu026 commented Mar 31, 2026

ultralytics包已经安装了,运行python examples/yolo/classify/run_train.py --save_dir ./runs/cls/train_finetune 出现下面的问题。readme里最好提供下pt权重的下载链接。

Traceback (most recent call last):
File "/home/mindspore/work/mindnlp/src/mindnlp/ultralytics/examples/yolo/classify/run_train.py", line 14, in
from ultralytics.models.yolo.classify.train import ClassificationTrainer
File "/home/mindspore/work/mindnlp/src/mindnlp/ultralytics/init.py", line 1, in
from .models.model import YOLO
File "/home/mindspore/work/mindnlp/src/mindnlp/ultralytics/models/init.py", line 1, in
from .model import YOLO
File "/home/mindspore/work/mindnlp/src/mindnlp/ultralytics/models/model.py", line 13, in
from ultralytics.models.yolo.classify.train import ClassificationTrainer
File "/home/mindspore/work/mindnlp/src/mindnlp/ultralytics/models/yolo/classify/train.py", line 10, in
from ultralytics.models.yolo.classify.val import ClassificationValidator
File "/home/mindspore/work/mindnlp/src/mindnlp/ultralytics/models/yolo/classify/val.py", line 8, in
from ultralytics.data.loaders import create_dataloader
ModuleNotFoundError: No module named 'ultralytics.data'

@kittentruck
Copy link
Copy Markdown
Author

已添加data文件夹,并在readme中补充上了权重下载链接

@moyu026
Copy link
Copy Markdown
Contributor

moyu026 commented Mar 31, 2026

MindNLP 0.6.0 版本下Classify,Detect,Segment,Pose任务都可以运行

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants