找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
查看: 3188|回复: 1
打印 上一主题 下一主题
收起左侧

给予SVM的手写数字识别上位机(含源码)VS2010 + OpenCV2.4.9开发

[复制链接]
跳转到指定楼层
楼主
自己写了一个基于SVM的数字手写识别分享给大家;源码见附件, VS2010 + OpenCV2.4.9开发
识别效果如下图所示:


VS2010源程序如下:

  1. // HandWriteDlg.cpp : 实现文件
  2. //

  3. #include "stdafx.h"
  4. #include "HandWrite.h"
  5. #include "HandWriteDlg.h"
  6. #include "afxdialogex.h"
  7. #include "opencv2/opencv.hpp"
  8. #include "windows.h"
  9. #include "fstream"
  10. #include <vector>

  11. //#pragma comment(lib,"ml.lib")

  12. using namespace std;
  13. using namespace cv;  

  14. #ifdef _DEBUG
  15. #define new DEBUG_NEW
  16. #endif


  17. // 用于应用程序“关于”菜单项的 CAboutDlg 对话框

  18. class CAboutDlg : public CDialogEx
  19. {
  20. public:
  21.         CAboutDlg();

  22. // 对话框数据
  23.         enum { IDD = IDD_ABOUTBOX };

  24.         protected:
  25.         virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

  26. // 实现
  27. protected:
  28.         DECLARE_MESSAGE_MAP()
  29. };

  30. CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
  31. {
  32. }

  33. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  34. {
  35.         CDialogEx::DoDataExchange(pDX);
  36. }

  37. BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
  38. END_MESSAGE_MAP()


  39. // CHandWriteDlg 对话框




  40. CHandWriteDlg::CHandWriteDlg(CWnd* pParent /*=NULL*/)
  41.         : CDialogEx(CHandWriteDlg::IDD, pParent)
  42.         , m_strRes(_T(""))
  43. {
  44.         m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  45. }

  46. void CHandWriteDlg::DoDataExchange(CDataExchange* pDX)
  47. {
  48.         CDialogEx::DoDataExchange(pDX);
  49.         DDX_Text(pDX, IDC_EDIT_RESULT, m_strRes);
  50. }

  51. BEGIN_MESSAGE_MAP(CHandWriteDlg, CDialogEx)
  52.         ON_WM_SYSCOMMAND()
  53.         ON_WM_PAINT()
  54.         ON_WM_QUERYDRAGICON()
  55.         ON_WM_MOUSEMOVE()
  56.         ON_WM_LBUTTONUP()
  57.         ON_WM_LBUTTONDOWN()
  58.         ON_BN_CLICKED(IDC_BUTTON_ERASE, &CHandWriteDlg::OnBnClickedButtonErase)
  59.         ON_BN_CLICKED(IDC_BUTTON_IDENTIFY, &CHandWriteDlg::OnBnClickedButtonIdentify)
  60. END_MESSAGE_MAP()


  61. // CHandWriteDlg 消息处理程序

  62. BOOL CHandWriteDlg::OnInitDialog()
  63. {
  64.         CDialogEx::OnInitDialog();

  65.         // 将“关于...”菜单项添加到系统菜单中。

  66.         // IDM_ABOUTBOX 必须在系统命令范围内。
  67.         ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  68.         ASSERT(IDM_ABOUTBOX < 0xF000);

  69.         CMenu* pSysMenu = GetSystemMenu(FALSE);
  70.         if (pSysMenu != NULL)
  71.         {
  72.                 BOOL bNameValid;
  73.                 CString strAboutMenu;
  74.                 bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
  75.                 ASSERT(bNameValid);
  76.                 if (!strAboutMenu.IsEmpty())
  77.                 {
  78.                         pSysMenu->AppendMenu(MF_SEPARATOR);
  79.                         pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  80.                 }
  81.         }

  82.         // 设置此对话框的图标。当应用程序主窗口不是对话框时,框架将自动
  83.         //  执行此操作
  84.         SetIcon(m_hIcon, TRUE);                        // 设置大图标
  85.         SetIcon(m_hIcon, FALSE);                // 设置小图标

  86.         // TODO: 在此添加额外的初始化代码

  87.         down_flag = false;
  88.         m_firstFlag = false;
  89.         m_start.x = 0;
  90.         m_start.y = 0;
  91.         m_start.x = 0;
  92.         m_start.y = 0;
  93.         CWnd* pWnd = GetDlgItem(IDC_STATIC_HAND);
  94.         pWnd->GetClientRect(&m_rect); //获取Picture控件的屏幕坐标区域   
  95.        
  96.         //this->ScreenToClient(&m_rect);


  97.         return TRUE;  // 除非将焦点设置到控件,否则返回 TRUE
  98. }

  99. void CHandWriteDlg::OnSysCommand(UINT nID, LPARAM lParam)
  100. {
  101.         if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  102.         {
  103.                 CAboutDlg dlgAbout;
  104.                 dlgAbout.DoModal();
  105.         }
  106.         else
  107.         {
  108.                 CDialogEx::OnSysCommand(nID, lParam);
  109.         }
  110. }

  111. // 如果向对话框添加最小化按钮,则需要下面的代码
  112. //  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序,
  113. //  这将由框架自动完成。

  114. void CHandWriteDlg::OnPaint()
  115. {
  116.         if (IsIconic())
  117.         {
  118.                 CPaintDC dc(this); // 用于绘制的设备上下文

  119.                 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

  120.                 // 使图标在工作区矩形中居中
  121.                 int cxIcon = GetSystemMetrics(SM_CXICON);
  122.                 int cyIcon = GetSystemMetrics(SM_CYICON);
  123.                 CRect rect;
  124.                 GetClientRect(&rect);
  125.                 int x = (rect.Width() - cxIcon + 1) / 2;
  126.                 int y = (rect.Height() - cyIcon + 1) / 2;

  127.                 // 绘制图标
  128.                 dc.DrawIcon(x, y, m_hIcon);
  129.         }
  130.         else
  131.         {
  132.                 CDialogEx::OnPaint();
  133.         }
  134. }

  135. //当用户拖动最小化窗口时系统调用此函数取得光标
  136. //显示。
  137. HCURSOR CHandWriteDlg::OnQueryDragIcon()
  138. {
  139.         return static_cast<HCURSOR>(m_hIcon);
  140. }



  141. void CHandWriteDlg::OnMouseMove(UINT nFlags, CPoint point)
  142. {
  143.         // TODO: 在此添加消息处理程序代码和/或调用默认值
  144.         if (!PtInRect(m_rect,CPoint(point)))
  145.         {
  146.                 down_flag=false;
  147.         }
  148.         CWnd* pWnd = GetDlgItem(IDC_STATIC_HAND);
  149.         CDC* dc = pWnd->GetDC();
  150.         CPen pen(PS_SOLID,10,RGB(0,0,0));
  151.         CPen *pOldPen=dc->SelectObject(&pen); //将其选入设备表
  152.         if (down_flag)
  153.         {
  154.                 dc->MoveTo(m_start);                     /* 开始画图 */
  155.                 dc->LineTo(point);
  156.                 m_start=point;
  157.         }
  158.         dc->SelectObject(pOldPen);

  159.         ReleaseDC(dc);
  160.         CDialogEx::OnMouseMove(nFlags, point);
  161. }


  162. void CHandWriteDlg::OnLButtonUp(UINT nFlags, CPoint point)
  163. {
  164.         // TODO: 在此添加消息处理程序代码和/或调用默认值
  165.         down_flag=false;
  166.         CDialogEx::OnLButtonUp(nFlags, point);
  167. }


  168. void CHandWriteDlg::OnLButtonDown(UINT nFlags, CPoint point)
  169. {
  170.         // TODO: 在此添加消息处理程序代码和/或调用默认值
  171.         if (PtInRect(m_rect,CPoint(point)))
  172.         {               
  173.                 down_flag=true;
  174.                 m_firstFlag = true;
  175.                 m_start=point;       
  176.         }

  177.         CDialogEx::OnLButtonDown(nFlags, point);
  178. }


  179. void CHandWriteDlg::OnBnClickedButtonErase()
  180. {
  181.         // TODO: 在此添加控件通知处理程序代码
  182.         m_firstFlag = false;
  183.         Invalidate();
  184. }


  185. void CHandWriteDlg::OnBnClickedButtonIdentify()
  186. {
  187.         // TODO: 在此添加控件通知处理程序代码
  188.         if(m_firstFlag == false)
  189.         {
  190.                 MessageBox(_T("请现在手写区写入数字!"));
  191.                 return ;
  192.         }
  193.        
  194.         CvSVM svm;
  195.         IplImage *test;
  196.         CString strResult;
  197.         char* str;
  198.         CRect pRect;
  199.         CWnd* pwnd=GetDlgItem(IDC_STATIC_HAND);//获取静态控件的指针 CRect pRect;
  200.         CDC* pDC=pwnd->GetDC(); //使用控件指针创建绘图用的DC
  201.         pwnd->GetClientRect(&pRect);
  202.         int w, h;
  203.         w = pRect.right - pRect.left;
  204.         h = pRect.bottom - pRect.top;
  205.         CBitmap bm;
  206.         bm.CreateCompatibleBitmap(pDC, w, h);
  207.         CDC memdc;
  208.         memdc.CreateCompatibleDC(pDC);
  209.         CBitmap*pOld=memdc.SelectObject(&bm);
  210.         memdc.BitBlt( 0, 0, w, h, pDC, pRect.left, pRect.top, SRCCOPY );
  211.         BITMAP btm;
  212.         bm.GetBitmap(&btm);
  213.         DWORD size=btm.bmWidthBytes*btm.bmHeight;
  214.         LPSTR lpData=(LPSTR)::GlobalAlloc(GPTR,size);
  215.         BITMAPINFOHEADER bih;
  216.         bih.biBitCount=btm.bmBitsPixel;
  217.         bih.biClrImportant=0;
  218.         bih.biClrUsed=0;
  219.         bih.biCompression=0;
  220.         bih.biHeight=btm.bmHeight;
  221.         bih.biPlanes=1;
  222.         bih.biSize=sizeof(BITMAPINFOHEADER);
  223.         bih.biSizeImage=size;
  224.         bih.biWidth=btm.bmWidth;
  225.         bih.biXPelsPerMeter=0;
  226.         bih.biYPelsPerMeter=0;
  227.         GetDIBits(memdc,bm,0,bih.biHeight,lpData,(BITMAPINFO*)&bih,DIB_RGB_COLORS);
  228.         BITMAPFILEHEADER bfh;
  229.         bfh.bfReserved1=bfh.bfReserved2=0;
  230.         bfh.bfType=((WORD)('M'<< 8)|'B');
  231.         bfh.bfSize=54+size;
  232.         bfh.bfOffBits=54;

  233.         CFile bf;
  234.         if(bf.Open(_T("a.bmp"),CFile::modeCreate|CFile::modeWrite))
  235.         {
  236.                 bf.Write(&bfh,sizeof(BITMAPFILEHEADER));
  237.                 bf.Write(&bih,sizeof(BITMAPINFOHEADER));
  238.                 bf.Write(lpData,size);
  239. ……………………

  240. …………限于本文篇幅 余下代码请从51黑下载附件…………
复制代码

所有资料51hei提供下载:
HandWrite.rar (15.41 MB, 下载次数: 38)




评分

参与人数 1黑币 +50 收起 理由
admin + 50 共享资料的黑币奖励!

查看全部评分

分享到:  QQ好友和群QQ好友和群 QQ空间QQ空间 腾讯微博腾讯微博 腾讯朋友腾讯朋友
收藏收藏 分享淘帖 顶 踩
回复

使用道具 举报

沙发
ID:418872 发表于 2018-11-1 01:13 | 只看该作者
学习了
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

手机版|小黑屋|51黑电子论坛 |51黑电子论坛6群 QQ 管理员QQ:125739409;技术交流QQ群281945664

Powered by 单片机教程网

快速回复 返回顶部 返回列表