opencv例子

Mat是OpenCV和C++的接口矩阵类,ImlImage是OpenCV和C语言的接口的结构体。
IplImage转Mat
//使用Mat的构造函数:Mat::Mat(const IplImage* img, bool copyData=false);    默认情况下copyData为false
IplImage* srcImg = cvLoadImage("Lena.jpg");
//注意:以下三种效果一致,均为浅拷贝
Mat M(srcImg);
Mat M(srcImg, false);
Mat M = srcImg;

//注意:当将参数copyData设为true后,则为深拷贝(复制整个图像数据)
Mat M(srcImg, true);

Mat转IplImage
//注意:浅拷贝 - 同样只是创建图像头,而没有复制数据
Mat M;
IplImage img = M;
IplImage img = IplImage(M);
//深拷贝
cv::Mat img2;
IplImage imgTmp = img2;
IplImage *input = cvCloneImage(&imgTmp);


例子有:显示YUV,DCT实现图像压缩。

显示YUV。

#include <opencv2/core/core.hpp>

#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <iostream>
#include <fstream>
#include <boost/filesystem.hpp>
#include <boost/thread.hpp>
#include "generalImgFun.hpp"
#include <math.h>

 using namespace std;
 using namespace cv;

 void WriteYuv()
 {
     cv::VideoCapture vc;
     bool flag = vc.open("S1000008.avi");
     if (!flag)
     {
         printf("avi file open error \n");
         system("pause");
         exit(-1);
     }

     int frmCount = vc.get(CV_CAP_PROP_FRAME_COUNT);
     frmCount -= 5;
     printf("frmCount: %d \n", frmCount);

     int w = vc.get(CV_CAP_PROP_FRAME_WIDTH);
     int h = vc.get(CV_CAP_PROP_FRAME_HEIGHT);
     int bufLen = w*h*3/2;
     unsigned char* pYuvBuf = new unsigned char[bufLen];
     FILE* pFileOut = fopen("result.yuv", "w+");
     if (!pFileOut)
     {
         printf("pFileOut open error \n");
         system("pause");
         exit(-1);
     }
     printf("pFileOut open ok \n");

     for (int i=0; i<frmCount; i++)
     {
         printf("%d/%d \n", i+1, frmCount);

         cv::Mat srcImg;
         vc>>srcImg;

         cv::imshow("img", srcImg);
         cv::waitKey(1);

         cv::Mat yuvImg;
         cv::cvtColor(srcImg, yuvImg, CV_BGR2YUV_I420);
         memcpy(pYuvBuf, yuvImg.data, bufLen*sizeof(unsigned char));

         fwrite(pYuvBuf, bufLen*sizeof(unsigned char), 1, pFileOut);
     }

     fclose(pFileOut);
     delete[] pYuvBuf;
 }

 void DisplayYUV(int w,int h , int fps ,string yuv_file_path)
 {
     printf("yuv file w: %d, h: %d \n", w, h);

     FILE* pFileIn = fopen(yuv_file_path.c_str(), "rb+");

     fseek(pFileIn, 0, SEEK_END);
     int frame_count = 0;

     frame_count = (int) ((int)ftell(pFileIn)/((w * h * 3) / 2));  // ftell 用于求文件大小
     printf("frame num is %d \n", frame_count);
     fseek(pFileIn, 0, SEEK_SET);//文件内位置定位到文件头

     int bufLen = w*h*3/2;
     unsigned char* pYuvBuf = new unsigned char[bufLen];

     for(int i=0; i<frame_count; i++)
     {
         fread(pYuvBuf, bufLen*sizeof(unsigned char), 1, pFileIn);

         cv::Mat yuvImg;
         yuvImg.create(h*3/2, w, CV_8UC1);
         memcpy(yuvImg.data, pYuvBuf, bufLen*sizeof(unsigned char));
         cv::Mat rgbImg;
         cv::cvtColor(yuvImg, rgbImg, CV_YUV2BGR_I420);

//         cv::imshow("yuv", yuvImg);
         cv::imshow("rgb", rgbImg);
         cv::waitKey(1000/fps);

         printf("cnt: %d \n", i);
     }

     delete[] pYuvBuf;


     fclose(pFileIn);
 }

int main(int argc, char** argv)
{
    if(argc<4){
        cout<<"usage: ./yuv_checker yuv_file.yuv width height fps!"<<endl;
        exit(1);
    }
    stringstream ss; ss <<argv[2];
    int nWidth ;
    int nHeight;
    int fps ;

    ss >> nWidth ;
    ss.clear(); ss<<argv[3];
    ss >> nHeight ;
    ss.clear(); ss<<argv[4];
    ss >> fps ;

    cout <<" width:"<<argv[2]<<" height:"<<argv[3]<<" fps:"<<argv[4]<<" file:"<<argv[1]<<endl;

    DisplayYUV(nWidth,nHeight,fps,argv[1]);

    return 0;
}

DCT实现图像压缩

#include "highgui.h"
#include <math.h>
#include <cv.h>
#include "cxcore.h"
#define cvCvtPlaneToPix cvMerge
double PSNR_B = 0;
double PSNR_G = 0;
double PSNR_R = 0;
double PSNR;

int main(int argc, char* argv[])
{
    const char* imagename = "F:/c/59.jpg";
    IplImage *src;
    CvScalar SrcPixel;
    CvScalar DstPixel;
    double sumB = 0;
    double sumG = 0;
    double sumR = 0;
    double mseB;
    double mseG;
    double mseR;

    src = cvLoadImage(imagename, 1);
    if (!src)
    {
        printf("can't open the image...\n");
        return -1;
    }
    // YUV颜色空间   
    IplImage* YUVImage = cvCreateImage(cvSize(src->width, src->height), src->depth, 3);
    IplImage* dst = cvCreateImage(cvSize(src->width, src->height), src->depth, 3);
    // YUV颜色空间各通道   
    IplImage* Y = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U, 1);
    IplImage* U = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U, 1);
    IplImage* V = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U, 1);

    //cvNamedWindow( "Origin Image", CV_WINDOW_AUTOSIZE );   
    cvCvtColor(src, YUVImage, CV_BGR2YUV); //BGR→YUV   
    cvSplit(YUVImage, Y, U, V, NULL);//分割通道   

    CvMat* MatY = cvCreateMat(Y->height, Y->width, CV_64FC1);
    CvMat* MatU = cvCreateMat(V->height, U->width, CV_64FC1);
    CvMat* MatV = cvCreateMat(V->height, V->width, CV_64FC1);

    CvMat* DCTY = cvCreateMat(Y->height, Y->width, CV_64FC1);
    CvMat* DCTU = cvCreateMat(U->height, U->width, CV_64FC1);
    CvMat* DCTV = cvCreateMat(V->height, V->width, CV_64FC1);

    cvScale(Y, MatY);
    cvScale(U, MatU);
    cvScale(V, MatV);

    cvDCT(MatY, DCTY, CV_DXT_FORWARD); //余弦变换   
    cvDCT(MatU, DCTU, CV_DXT_FORWARD); //余弦变换   
    cvDCT(MatV, DCTV, CV_DXT_FORWARD); //余弦变换   

                                       //Y 通道压缩   
    for (int i = 0; i < Y->height; i++)
    {
        for (int j = 0; j < Y->width; j++)
        {
            double  element = CV_MAT_ELEM(*DCTY, double, i, j);
            if (abs(element) < 10)
                CV_MAT_ELEM(*DCTY, double, i, j) = 0;
        }
    }

    // U 通道压缩   
    for (int i = 0; i < U->height; i++)
    {
        for (int j = 0; j < U->width; j++)
        {
            double  element = CV_MAT_ELEM(*DCTU, double, i, j);
            if (abs(element) < 20)
                CV_MAT_ELEM(*DCTU, double, i, j) = 0;
        }
    }

    // V 通道压缩   
    for (int i = 0; i < V->height; i++)
    {
        for (int j = 0; j < V->width; j++)
        {
            double  element = CV_MAT_ELEM(*DCTV, double, i, j);
            if (abs(element) < 20)
                CV_MAT_ELEM(*DCTV, double, i, j) = 0;
        }
    }
    cvDCT(DCTY, MatY, CV_DXT_INVERSE); //余弦反变换   
    cvDCT(DCTU, MatU, CV_DXT_INVERSE);
    cvDCT(DCTV, MatV, CV_DXT_INVERSE);

    cvScale(MatY, Y);
    cvScale(MatU, U);
    cvScale(MatV, V);

    cvMerge(Y, U, V, NULL, YUVImage);
    cvCvtColor(YUVImage, dst, CV_YUV2BGR); //YUV→BGR   

                                           //  计算前后两幅图像的PSNR值   
    for (int i = 0; i < src->height; i++)
    {
        for (int j = 0; j < src->width; j++)
        {
            SrcPixel = cvGet2D(src, i, j);
            DstPixel = cvGet2D(dst, i, j);
            sumB += (SrcPixel.val[0] - DstPixel.val[0]) * (SrcPixel.val[0] - DstPixel.val[0]);
            sumG += (SrcPixel.val[1] - DstPixel.val[1]) * (SrcPixel.val[1] - DstPixel.val[1]);
            sumR += (SrcPixel.val[2] - DstPixel.val[2]) * (SrcPixel.val[2] - DstPixel.val[2]);

        }
    }
    mseB = sumB / ((src->width) * (src->height)); //计算均方差   
    mseG = sumG / ((src->width) * (src->height));
    mseR = sumR / ((src->width) * (src->height));

    PSNR_B = 10.0 * (log10(255.0 * 255.0 / mseB));
    PSNR_G = 10.0 * (log10(255.0 * 255.0 / mseG));
    PSNR_R = 10.0 * (log10(255.0 * 255.0 / mseR));
    PSNR = (PSNR_B + PSNR_G + PSNR_R) / 3;
    printf("PSNR:%d ", PSNR_B);
    cvShowImage("YImage", Y);
    cvShowImage("UImage", U);
    cvShowImage("VImage", V);
    cvShowImage("DstImage", dst);
    cvSaveImage("F:/dstdemo.jpg", dst);

    while (1)
    {
        if (cvWaitKey(0) == 27) break;
    }

    cvDestroyWindow("YImage");
    cvDestroyWindow("UImage");
    cvDestroyWindow("VImage");
    cvDestroyWindow("DstImage");


    cvReleaseImage(&Y);
    cvReleaseImage(&U);
    cvReleaseImage(&V);
    cvReleaseImage(&src);
    cvReleaseImage(&dst);
    cvReleaseImage(&YUVImage);
    system("pause");
    return 0;

}






评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值