软件功能

如何在灵闪的脚本模块调用第三方python库

返回案例列表

问题描述

如何在灵闪软件的脚本模块中调用第三方Python库?

原始问题图片

如何在灵闪的脚本模块调用第三方python库?

灵闪的python脚本工具原理上是支持python社区所有第三方包的调用的,但目前暂没有命令行可以直接安装python包

解答

通过sys.path.append()可以将第三方库所在的路径给添加到python的库搜索路径中。

调用YOLO代码

import sys

# 添加pytorch等库所在的路径
sys.path.append(r'C:\Users\abc\.conda\envs\mt11.7\Lib\site-packages')

# 添加yolo库所在的路径
sys.path.append(r'D:\ib_project\ark\model_zoo\ultralytics')

# 导入yolo的库
import ultralytics
from ultralytics import YOLO

# 初始化模型
model = YOLO(r"D:\ib_project\ark\yolov8n.pt")

# 模型推理图片
results = model([图像增强.输出图像])

# 绘制推理结果
draw_res = results[0].plot()