تبدیل cv::Mat به tensor در تسورفلو

Question

سلام.

من این پست را دیدم که قبلاً جواب داده شده بود ولی من تو c++ می خوام تصویر cv::Mat را به تنسور تبدیل کنم.

 

ممنون

Answer 1

به صورت زیر:

Mat frame;
frame=imread("./cara.jpg");
Tensor inputImg(tensorflow::DT_UINT8, tensorflow::TensorShape({1,frame.rows,frame.cols,3}));
uint8_t *p = inputImg.flat<tensorflow::uint8>().data();
Mat cameraImg(frame.rows, frame.cols, CV_8UC3, p);
frame.convertTo(cameraImg, CV_8UC3);

 

Answer 2

چون اغلب شبکه ها تصاویر ورودی آنها به صورت اعشاری ست بهتر بود کد @roshanak به صورت اغشاری نوشته می شد :

Mat frame;
frame=imread("./cara.jpg");
Tensor inputImg(tensorflow::DT_FLOAT, tensorflow::TensorShape({1,frame.rows,frame.cols,3}));
float *p = inputImg.flat<float>().data();
Mat cameraImg(frame.rows, frame.cols, CV_32FC3, p);
frame.convertTo(cameraImg, CV_32FC3);