sopt/cppflow__utils_8h_source.html

 #ifndef SOPT_CPPFLOWUTILS_H

 #define SOPT_CPPFLOWUTILS_H


 #include <Eigen/Core>

 #include "sopt/types.h"

 #include <cppflow/cppflow.h>

 #include "cppflow/ops.h"

 #include <complex>

 #include <vector>


 namespace sopt::cppflowutils {


 cppflow::tensor convert_image_to_tensor(sopt::Image<double> const &image, int image_rows, int image_cols);

 cppflow::tensor convert_image_to_tensor(sopt::Image<std::complex<double>> const &image, int image_rows, int image_cols);

 cppflow::tensor convert_image_to_tensor(sopt::Vector<double> const &image, int image_rows, int image_cols);

 cppflow::tensor convert_image_to_tensor(sopt::Vector<std::complex<double>> const &image, int image_rows, int image_cols);


 Eigen::Map<Eigen::Array<double, Eigen::Dynamic, Eigen::Dynamic>> convert_tensor_to_image(std::vector<float> model_output, int image_rows, int image_cols);


 } // namespace sopt::cppflowutils

 #endif

sopt::cppflowutils
Definition: cppflow_utils.cc:9

sopt::cppflowutils::convert_image_to_tensor
cppflow::tensor convert_image_to_tensor(Image< double > const &image, int image_rows, int image_cols)
Converts a sopt::Image to a cppflow::tensor.
Definition: cppflow_utils.cc:14

sopt::cppflowutils::convert_tensor_to_image
Eigen::Map< Eigen::Array< double, Eigen::Dynamic, Eigen::Dynamic > > convert_tensor_to_image(std::vector< float > model_output, int image_rows, int image_cols)
Convert a cppflow:tensor to an Eigen::Array.
Definition: cppflow_utils.cc:86

sopt::Image
Eigen::Array< T, Eigen::Dynamic, Eigen::Dynamic > Image
A 2-dimensional list of elements of given type.
Definition: types.h:39

sopt::Vector
Eigen::Matrix< T, Eigen::Dynamic, 1 > Vector
A vector of a given type.
Definition: types.h:24

types.h