opencv C++图像风格迁移 网络模型文件 及源代码下载

#include <opencv2/opencv.hpp> #include <iostream> using namespace cv; using namespace cv::dnn; using namespace std; int main() { Mat image = imread("cat2.png"); String models[5] = { "the_wave.t7", "mosaic.t7", "feathers.t7", "candy.t7", "udnie.t7" }; for (int i = 0; i < size(models); i++) { Net net = readNet("fast_style/" + models[i]); imshow("原始图像", image); //计算图像每个通道的均值 Scalar imgaeMean = mean(image); //调整图像尺寸和格式 Mat blobImage = blobFromImage(image, 1.0, Size(256, 256), imgaeMean, false, false); //计算网络对原图像处理结果 net.setInput(blobImage);//默认参数，数据输入到模型第一层，即输入层。 Mat output = net.forward();//前向计算，没有输入参数，默认一直计算到底，得到结果 //输出结果的尺寸和通道数 //输出结果，三个参数依次是通道数、行数、列数，分别取出。 int outputChannels = output.size[1]; int outputRows = output.size[2]; int outputCols = output.size[3]; //将输出结果存放到图像中 Mat result = Mat::zeros(Size(outputCols, outputRows), CV_32FC3); //创建一个与输出结果尺寸相同，三通道的全0图像 float* data = output.ptr<float>(); for (int channel = 0; channel < outputChannels; channel++) { for (int row = 0; row < outputRows; row++) { for (int col = 0; col < outputCols; col++) { result.at<Vec3f>(row, col)[channel] = *data++;//将输出结果 对应行和列的数据放入到 新的图像中 } } } //对迁移结果进行进一步操作处理 //加上之前减掉的均值 result = result + imgaeMean; //对图像进行归一化，便于图像显示 result = result / 255.0; //调整图像尺寸，使得与原图像尺寸相同 resize(result, result, image.size()); //显示结果 imshow("第" + to_string(i) + "种风格迁移结果", result); } waitKey(0); return 0; }