我用Cython封装了一个来自segment-anything-master项目里的.pth模型,我的.pyx文件是这么写的import sys import numpy as np import matplotlib import matplotlib.pyplot as plt import cv2 from typing import Any, Dict, List, Optional, Tuple sys.path.append("/app/PythonProjects/segment-anything-main") from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor # model.pyxcdef class SamModel: def __init__(self, str model_type,str sam_checkpoint, str device = "cpu", points_per_side: Optional[int] = 32, points_per_batch: int = 64, point_grids: Optional[List[np.ndarray]] = None): self.sam = sam_model_registry[model_type](checkpoint=sam_checkpoint) self.sam.to(device) self.mask_generator = SamAutomaticMaskGenerator(self.sam,points_per_side = points_per_side, points_per_batch = points_per_batch, point_grids = point_grids) def __getmodel__(self): return self.sam def __callmodel__( self, input: np.array ): masks = self.mask_generator.generate(input) return masks,我的setup.py文件是这样写的:from distutils.core import setup from distutils.extension import Extension import Cython from Cython.Build import cythonize import numpy extensions = [ Extension('models.model', ['/app/PythonProjects/segment-anything-main/SamCython/models/model.pyx'], include_dirs=[numpy.get_include()]), ] setup( name='models', ext_modules=cythonize(extensions), ),给我一篇C++调动这个模型的实例

时间: 2023-12-27 16:02:24 浏览: 213
好的,以下是一个简单的C++调用这个模型的示例代码: ```cpp #include <Python.h> #include <numpy/arrayobject.h> #include <iostream> int main() { // 初始化 Python 解释器 Py_Initialize(); // 加载 NumPy 模块 import_array(); // 加载 Python 模块和函数 PyObject* pModule = PyImport_ImportModule("models.model"); if (!pModule) { std::cerr << "Failed to import module" << std::endl; return 1; } PyObject* pFunc = PyObject_GetAttrString(pModule, "__callmodel__"); if (!pFunc || !PyCallable_Check(pFunc)) { std::cerr << "Failed to get function" << std::endl; return 1; } // 准备输入数据 int width = 640; int height = 480; npy_intp dims[3] = { height, width, 3 }; unsigned char* inputData = new unsigned char[width * height * 3]; // 填充 inputData 数据 // 创建 NumPy 数组 PyObject* pInputArray = PyArray_SimpleNewFromData( 3, dims, NPY_UINT8, inputData); // 调用函数 PyObject* pOutputArray = PyObject_CallFunction(pFunc, "O", pInputArray); if (!pOutputArray) { std::cerr << "Failed to call function" << std::endl; return 1; } // 获取输出数据 unsigned char* outputData = PyArray_DATA((PyArrayObject*)pOutputArray); // 处理 outputData 数据 // 释放资源 Py_DECREF(pInputArray); Py_DECREF(pOutputArray); Py_DECREF(pFunc); Py_DECREF(pModule); Py_Finalize(); delete[] inputData; return 0; } ``` 需要注意的是,在编译时,需要链接 Python 和 NumPy 库。可以使用以下命令进行编译: ``` g++ -o main main.cpp -I/usr/include/python3.8 -lpython3.8 -lpython3 -lm -ldl -I/usr/local/lib/python3.8/dist-packages/numpy/core/include -L/usr/local/lib/python3.8/config-3.8-x86_64-linux-gnu -lpython3.8 -lcrypt -lpthread -ldl -lutil -lm -Wno-unused-result ``` 其中,`-I` 参数指定 Python 和 NumPy 的头文件路径,`-l` 参数指定 Python 的库文件和 NumPy 的库文件(需要根据自己的 Python 版本和安装路径进行调整)。
阅读全文

相关推荐

wandb: Tracking run with wandb version 0.15.5 wandb: W&B syncing is set to offline in this directory. wandb: Run wandb online or set WANDB_MODE=online to enable cloud syncing. /home/zhangmengjie/anaconda3/envs/torch1/lib/python3.7/site-packages/gym/envs/registration.py:556: UserWarning: WARN: The environment Ant-v2 is out of date. You should consider upgrading to version v4. f"The environment {id} is out of date. You should consider " Error compiling Cython file: ------------------------------------------------------------ ... See c_warning_callback, which is the C wrapper to the user defined function ''' global py_warning_callback global mju_user_warning py_warning_callback = warn mju_user_warning = c_warning_callback ^ ------------------------------------------------------------ /home/zhangmengjie/anaconda3/envs/torch1/lib/python3.7/site-packages/mujoco_py/cymj.pyx:92:23: Cannot assign type 'void (const char *) except * nogil' to 'void (*)(const char *) noexcept nogil' Error compiling Cython file: ------------------------------------------------------------ ... See c_warning_callback, which is the C wrapper to the user defined function ''' global py_error_callback global mju_user_error py_error_callback = err_callback mju_user_error = c_error_callback ^ ------------------------------------------------------------ /home/zhangmengjie/anaconda3/envs/torch1/lib/python3.7/site-packages/mujoco_py/cymj.pyx:127:21: Cannot assign type 'void (const char *) except * nogil' to 'void (*)(const char *) noexcept nogil' Compiling /home/zhangmengjie/anaconda3/envs/torch1/lib/python3.7/site-packages/mujoco_py/cymj.pyx because it changed. [1/1] Cythonizing /home/zhangmengjie/anaconda3/envs/torch1/lib/python3.7/site-packages/mujoco_py/cymj.pyx wandb: Waiting for W&B process to finish... (failed 1). wandb: You can sync this run to the cloud by running: wandb: wandb sync /home/zhangmengjie/PID/Python/ERL-Re2-main/wandb/offline-run-20230721_165346-awq1hazo wandb: Find logs at: ./wandb/offline-run-20230721_165346-awq1hazo/logs

最新推荐

recommend-type

如何使用Cython对python代码进行加密

首先,理解Cython的基本概念:Cython是Python的一个超集,它允许开发者使用类似Python的语法编写代码,但可以编译成C语言,从而获得接近C的运行效率。Cython的主要优势在于,它保留了Python的易读性和便捷性,同时...
recommend-type

为Matlab 用户提供丰富功能的工具集

Matlab 用户提供丰富功能的工具集。 从优点来看,其涵盖了诸多实用的功能模块。例如在计算机视觉和图像处理领域,可能包含了图像特征提取、目标检测、图像分割等常用算法的实现,这为相关领域的研究人员和开发者节省了大量的时间和精力,无需从头编写复杂的算法代码,提高了开发效率。工具包的文档相对较为详细,能够帮助用户快速上手并理解各个函数和工具的使用方法,降低了使用门槛。同时,它可能具有较好的兼容性,能够与 Matlab 的原生功能和其他常用工具箱协同工作,方便用户整合到自己的项目中。 1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。
recommend-type

StarModAPI: StarMade 模组开发的Java API工具包

资源摘要信息:"StarModAPI: StarMade 模组 API是一个用于开发StarMade游戏模组的编程接口。StarMade是一款开放世界的太空建造游戏,玩家可以在游戏中自由探索、建造和战斗。该API为开发者提供了扩展和修改游戏机制的能力,使得他们能够创建自定义的游戏内容,例如新的星球类型、船只、武器以及各种游戏事件。 此API是基于Java语言开发的,因此开发者需要具备一定的Java编程基础。同时,由于文档中提到的先决条件是'8',这很可能指的是Java的版本要求,意味着开发者需要安装和配置Java 8或更高版本的开发环境。 API的使用通常需要遵循特定的许可协议,文档中提到的'在许可下获得'可能是指开发者需要遵守特定的授权协议才能合法地使用StarModAPI来创建模组。这些协议通常会规定如何分发和使用API以及由此产生的模组。 文件名称列表中的"StarModAPI-master"暗示这是一个包含了API所有源代码和文档的主版本控制仓库。在这个仓库中,开发者可以找到所有的API接口定义、示例代码、开发指南以及可能的API变更日志。'Master'通常指的是一条分支的名称,意味着该分支是项目的主要开发线,包含了最新的代码和更新。 开发者在使用StarModAPI时应该首先下载并解压文件,然后通过阅读文档和示例代码来了解如何集成和使用API。在编程实践中,开发者需要关注API的版本兼容性问题,确保自己编写的模组能够与StarMade游戏的当前版本兼容。此外,为了保证模组的质量,开发者应当进行充分的测试,包括单人游戏测试以及多人游戏环境下的测试,以确保模组在不同的使用场景下都能够稳定运行。 最后,由于StarModAPI是针对特定游戏的模组开发工具,开发者在创建模组时还需要熟悉StarMade游戏的内部机制和相关扩展机制。这通常涉及到游戏内部数据结构的理解、游戏逻辑的编程以及用户界面的定制等方面。通过深入学习和实践,开发者可以利用StarModAPI创建出丰富多样的游戏内容,为StarMade社区贡献自己的力量。" 由于题目要求必须输出大于1000字的内容,上述内容已经满足此要求。如果需要更加详细的信息或者有其他特定要求,请提供进一步的说明。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

R语言数据清洗术:Poisson分布下的异常值检测法

![R语言数据清洗术:Poisson分布下的异常值检测法](https://ucc.alicdn.com/images/user-upload-01/img_convert/a12c695f8b68033fc45008ede036b653.png?x-oss-process=image/resize,s_500,m_lfit) # 1. R语言与数据清洗概述 数据清洗作为数据分析的初级阶段,是确保后续分析质量的关键。在众多统计编程语言中,R语言因其强大的数据处理能力,成为了数据清洗的宠儿。本章将带您深入了解数据清洗的含义、重要性以及R语言在其中扮演的角色。 ## 1.1 数据清洗的重要性
recommend-type

设计一个简易的Python问答程序

设计一个简单的Python问答程序,我们可以使用基本的命令行交互,结合字典或者其他数据结构来存储常见问题及其对应的答案。下面是一个基础示例: ```python # 创建一个字典存储问题和答案 qa_database = { "你好": "你好!", "你是谁": "我是一个简单的Python问答程序。", "你会做什么": "我可以回答你关于Python的基础问题。", } def ask_question(): while True: user_input = input("请输入一个问题(输入'退出'结束):")
recommend-type

PHP疫情上报管理系统开发与数据库实现详解

资源摘要信息:"本资源是一个PHP疫情上报管理系统,包含了源码和数据库文件,文件编号为170948。该系统是为了适应疫情期间的上报管理需求而开发的,支持网络员用户和管理员两种角色进行数据的管理和上报。 管理员用户角色主要具备以下功能: 1. 登录:管理员账号通过直接在数据库中设置生成,无需进行注册操作。 2. 用户管理:管理员可以访问'用户管理'菜单,并操作'管理员'和'网络员用户'两个子菜单,执行增加、删除、修改、查询等操作。 3. 更多管理:通过点击'更多'菜单,管理员可以管理'评论列表'、'疫情情况'、'疫情上报管理'、'疫情分类管理'以及'疫情管理'等五个子菜单。这些菜单项允许对疫情信息进行增删改查,对网络员提交的疫情上报进行管理和对疫情管理进行审核。 网络员用户角色的主要功能是疫情管理,他们可以对疫情上报管理系统中的疫情信息进行增加、删除、修改和查询等操作。 系统的主要功能模块包括: - 用户管理:负责系统用户权限和信息的管理。 - 评论列表:管理与疫情相关的评论信息。 - 疫情情况:提供疫情相关数据和信息的展示。 - 疫情上报管理:处理网络员用户上报的疫情数据。 - 疫情分类管理:对疫情信息进行分类统计和管理。 - 疫情管理:对疫情信息进行全面的增删改查操作。 该系统采用面向对象的开发模式,软件开发和硬件架设都经过了细致的规划和实施,以满足实际使用中的各项需求,并且完善了软件架设和程序编码工作。系统后端数据库使用MySQL,这是目前广泛使用的开源数据库管理系统,提供了稳定的性能和数据存储能力。系统前端和后端的业务编码工作采用了Thinkphp框架结合PHP技术,并利用了Ajax技术进行异步数据交互,以提高用户体验和系统响应速度。整个系统功能齐全,能够满足疫情上报管理和信息发布的业务需求。" 【标签】:"java vue idea mybatis redis" 从标签来看,本资源虽然是一个PHP疫情上报管理系统,但提到了Java、Vue、Mybatis和Redis这些技术。这些技术标签可能是误标,或是在资源描述中提及的其他技术栈。在本系统中,主要使用的技术是PHP、ThinkPHP框架、MySQL数据库、Ajax技术。如果资源中确实涉及到Java、Vue等技术,可能是前后端分离的开发模式,或者系统中某些特定模块使用了这些技术。 【压缩包子文件的文件名称列表】: CS268000_*** 此列表中只提供了单一文件名,没有提供详细文件列表,无法确定具体包含哪些文件和资源,但假设它可能包含了系统的源代码、数据库文件、配置文件等必要组件。
recommend-type

"互动学习:行动中的多样性与论文攻读经历"

多样性她- 事实上SCI NCES你的时间表ECOLEDO C Tora SC和NCESPOUR l’Ingén学习互动,互动学习以行动为中心的强化学习学会互动,互动学习,以行动为中心的强化学习计算机科学博士论文于2021年9月28日在Villeneuve d'Asq公开支持马修·瑟林评审团主席法布里斯·勒菲弗尔阿维尼翁大学教授论文指导奥利维尔·皮耶昆谷歌研究教授:智囊团论文联合主任菲利普·普雷教授,大学。里尔/CRISTAL/因里亚报告员奥利维耶·西格德索邦大学报告员卢多维奇·德诺耶教授,Facebook /索邦大学审查员越南圣迈IMT Atlantic高级讲师邀请弗洛里安·斯特鲁布博士,Deepmind对于那些及时看到自己错误的人...3谢谢你首先,我要感谢我的两位博士生导师Olivier和Philippe。奥利维尔,"站在巨人的肩膀上"这句话对你来说完全有意义了。从科学上讲,你知道在这篇论文的(许多)错误中,你是我可以依
recommend-type

R语言统计推断:掌握Poisson分布假设检验

![R语言数据包使用详细教程Poisson](https://media.cheggcdn.com/media/a2b/a2b4ee79-229c-4cfe-a3bc-e4766a05004e/phpYTlWxe) # 1. Poisson分布及其统计推断基础 Poisson分布是统计学中一种重要的离散概率分布,它描述了在固定时间或空间内发生某独立事件的平均次数的分布情况。本章将带领读者了解Poisson分布的基本概念和统计推断基础,为后续章节深入探讨其理论基础、参数估计、假设检验以及实际应用打下坚实的基础。 ```markdown ## 1.1 Poisson分布的简介 Poisson分
recommend-type

NX C++二次开发高亮颜色设置的方法

NX C++二次开发中,高亮颜色设置通常涉及到自定义用户界面(UI)组件的外观。以下是一些常见的方法来设置高亮颜色: 1. **使用Qt样式表(StyleSheet)**: 如果你使用的是Qt框架进行开发,可以通过设置样式表来改变控件的高亮颜色。例如,对于按钮,你可以这样设置: ```cpp button->setStyleSheet("QPushButton:hover {background-color: yellow;}"); ``` 这会将鼠标悬停在按钮上时的背景色设置为黄色。 2. **直接修改属性**: 对于某些控件,可以直接通过修改其属性来