torch::Tensor out_tensor = module.forward({ input_tensor }).toTensor();
auto img_output = out_tensor;
auto img_uint = img_output.toType(torch::kU8);
auto img = cv::Mat(input_image.rows, input_image.cols, CV_8UC3, img_uint.data_ptr());
cv::cvtColor(img, img, cv::COLOR_BGR2RGB);
std::cout << "Tensor converetd to Image!!" << std::endl;
cv::imwrite("./Images/new_img.jpg", img);
I can’t get the desired output. Instead I get a noisy grayscale image.