Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- /*
- Some function to convert images between different formats
- Author: Alperen AYDIN
- */
- #ifndef CONVERSION_H
- #define CONVERSION_H
- #include <opencv2/core/core.hpp>
- #include <opencv2/highgui/highgui.hpp>
- #include <eigen3/unsupported/Eigen/CXX11/Tensor>
- #include "tensorflow/core/public/session.h"
- #include "tensorflow/core/platform/env.h"
- #include "tensorflow/core/framework/tensor.h"
- #include <vector>
- Eigen::Tensor<float,4> convertImage_eigen(cv::Mat M);
- Eigen::Tensor<float,4> convertImage_eigen(std::vector<cv::Mat> L);
- tensorflow::Tensor convertImage(cv::Mat M);
- tensorflow::Tensor convertImage(std::vector<cv::Mat> L);
- cv::Mat convertTensor_unitary( Eigen::Tensor<float,4> T);
- std::vector<cv::Mat> convertTensor( Eigen::Tensor<float,4> T);
- cv::Mat convertTensor_unitary(tensorflow::Tensor T);
- std::vector<cv::Mat> convertTensor(tensorflow::Tensor T);
- #endif
Add Comment
Please, Sign In to add comment