c++调用python的混编程序（vs下调用pytorch的深度学习接口）
// WallTileCoreAlgorithm.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
#include "RV_GetBirckInfo_all.h"
#include
#include
#include
void getName(char *imgPath, char*imgName)
{
std::stack
int count = 0;
while (imgPath[count] != '.')
{
count++;
}
count--;
while (imgPath[count] != '\\')
{
tempName.push(imgPath[count]);
count--;
}
int i = 0;
while (!tempName.empty())
{
imgName[i] = tempName.top();
tempName.pop();
i++;
}
return;
}
//////////////////////
typedef struct DizhuanPointInfo
{
double _x;
double _y;
DizhuanPointInfo() :_x(-1), _y(-1) {};
}DizhuanPointInfo;
double convertPythonObjectToDouble(PyObject *obj)
{
double result = 0;
PyArg_Parse(obj, "d", &result);
return result;
}
//数据转换
void convertPythonListObjectToDizhuanData(PyObject *pRet, std::vector
{
_dizhuanInfos.clear();
if (!pRet)
{
return;
}
if (!PyList_Check(pRet))
{
return;
}
//dizhuan python returns lead to this strange code.
PyArrayObject *pObj = (PyArrayObject *)PyList_GetItem(pRet, 0);
if (!PyList_Check(pObj))
{
return;
}
int listSize = PyList_Size((PyObject*)pObj);
for (int i = 0; i < listSize; i++)
{
PyArrayObject *currentPoint = (PyArrayObject *)PyList_GetItem((PyObject*)pObj, i);
if (!PyList_Check(currentPoint))
{
continue;
}
int sz = PyList_Size((PyObject*)currentPoint);
if (sz < 2)
{
continue;
}
//
PyObject *p0 = PyList_GetItem((PyObject *)currentPoint, 0);
PyObject *p1 = PyList_GetItem((PyObject *)currentPoint, 1);
//
auto d0 = convertPythonObjectToDouble(p0);
auto d1 = convertPythonObjectToDouble(p1);
DizhuanPointInfo tempInfo;
tempInfo._x = d0;
tempInfo._y = d1;
_dizhuanInfos.push_back(tempInfo);
}
}
void *test()
{
//初始化python模块
Py_SetPythonHome(L"C:\\Users\\dell\\Anaconda3\\envs\\pytorch");//add by wk_zhong
Py_Initialize();
import_array();
if (!Py_IsInitialized())
{
exit(1);
}
PyRun_SimpleString("import sys");
PyRun_SimpleString("sys.path.append('E:/Application_of_algorithm_project/DiZhuan_Robot/VisualAlgorithmProgram/FloorTileCoreAlgorithm/dizhuan_wmx/dizhuan')");
//PyRun_SimpleString("import evaluation-gai");
PyObject * pModule = nullptr;
pModule = PyImport_ImportModule("evaluationgai");
string _pvFuncName;
_pvFuncName = "evaluate2";
PyObject *_pv = nullptr;
_pv = PyObject_GetAttrString(pModule, _pvFuncName.c_str());
std::vector
if (pModule == nullptr)
{
std::cout << "cannot find wkzhong" << std::endl;
return nullptr;
}
char path[500];
char savePath[500];
char lebelname[500];
char tempPath[500];
///
FILE *fileIndex;
FILE *fileLabelName;
fileIndex = fopen(".\\index\\all_New.txt", "r");
float numOfPic = 0;
fscanf(fileIndex, "%f", &numOfPic);
for (int i = 0; i < numOfPic; i++)
{
std::cout << i / numOfPic << std::endl;
for (int i = 0; i < 500; i++)
{
lebelname[i] = 0;
}
std::cout << "Process Pic :" << i + 1 << std::endl;
fscanf(fileIndex, "%s", path);
getName(path, lebelname);
cv::Mat input = cv::imread(path);
//cv::Mat input = cv::imread("G:\\ProjectPic\\dizhuan\\待整理\\new-030405\\0520\\oriImg\\c52_20200516_132852.bmp");
if (input.empty())
return 0;
clock_t start, end, start1, end1;
start1 = clock();
cv::Mat ProcesImg;
RV_ImageProcess(input, ProcesImg);
//end1 = clock();
//std::cout << "total time:" << 1000 * (end1 - start1) / CLOCKS_PER_SEC << "ms" << std::endl;
RVBRICK brick1;
RVBRICK brick2;
RVBRICK brick3;
RVBRICK brick4;
cv::Mat dlimg; //传入DL算法的图片
bool isRotated = RV_DLpreprocess(input, dlimg); //DL算法预处理
if (dlimg.channels() == 1)
cv::cvtColor(dlimg, dlimg, cv::COLOR_GRAY2BGR); //图像必须是三通道的:612*512
//为接入深度学习，做数据转换
npy_intp imageShape[1] = { 512 * 612 * 3 };
PyByteArrayObject *pyImgArr;
pyImgArr = reinterpret_cast
PyObject *args = PyTuple_New(4);
PyObject *arg1 = Py_BuildValue("i", 612);
PyObject *arg2 = Py_BuildValue("i", 512);
PyObject *arg3 = Py_BuildValue("i", 3);
PyTuple_SetItem(args, 0, reinterpret_cast
PyTuple_SetItem(args, 1, arg1);
PyTuple_SetItem(args, 2, arg2);
PyTuple_SetItem(args, 3, arg3);
PyObject *pRet = PyObject_CallObject(_pv, args);
convertPythonListObjectToDizhuanData(pRet, DLResult);
Py_DecRef(pRet);
RV_DLResultPoint roughPt; //DL算法得到的数据
RV_DLResultPoint mapPos; //位置映射后的数据
//此处模拟dl得到后的结果赋值
if (DLResult.size() == 4)
{
roughPt.quadrant1.x = DLResult.at(0)._x;
roughPt.quadrant1.y = DLResult.at(0)._y;
roughPt.quadrant2.x = DLResult.at(1)._x;
roughPt.quadrant2.y = DLResult.at(1)._y;
roughPt.quadrant3.x = DLResult.at(2)._x;
roughPt.quadrant3.y = DLResult.at(2)._y;
roughPt.quadrant4.x = DLResult.at(3)._x;
roughPt.quadrant4.y = DLResult.at(3)._y;
cout << roughPt.quadrant1.x << endl;
}
else
{
roughPt.quadrant1 = cv::Point(-1, -1);
roughPt.quadrant2 = cv::Point(-1, -1);
roughPt.quadrant3 = cv::Point(-1, -1);
roughPt.quadrant4 = cv::Point(-1, -1);
}
RV_mappingPosition(roughPt, mapPos, isRotated); //位置映射函数
brick1.RV_GetBirckInfo(ProcesImg, 1, mapPos); //接入传统视觉算法
brick2.RV_GetBirckInfo(ProcesImg, 2, mapPos); //接入传统视觉算法
brick3.RV_GetBirckInfo(ProcesImg, 3, mapPos); //接入传统视觉算法
brick4.RV_GetBirckInfo(ProcesImg, 4, mapPos); //接入传统视觉算法
end1 = clock();
std::cout << "total time:" << 1000 * (end1 - start1) / CLOCKS_PER_SEC << "ms" << std::endl << std::endl;
//cv::cvtColor(input, input, cv::COLOR_GRAY2BGR);
cv::circle(input, brick1.crossPoint, 10, cv::Scalar(0, 0, 255), 3, 8);
cv::circle(input, brick2.crossPoint, 10, cv::Scalar(0, 255, 0), 3, 8);
cv::circle(input, brick3.crossPoint, 10, cv::Scalar(255, 0, 0), 3, 8);
cv::circle(input, brick4.crossPoint, 10, cv::Scalar(0, 255, 255), 3, 8);
std::sprintf(savePath, "G:\\ProjectPic\\dizhuan\\待整理\\DLResult345\\%s.jpg", lebelname);
cv::imwrite(savePath, input);
}
return nullptr;
}
int main()
{
test();
return 1;
}