使用openCV + python
1. 进入\OpenCV\interfaces\swig\python目录,执行
python setup-for-win.py install
进行安装
2.安装成功后,会发现在Python26\Lib\site-packages下多了opencv这一目录,里面应该有这些文件
adaptors.py
adaptors.pyc
cn.bmp
cn.bmp.jpg
cv.py
cv.pyc
highgui.py
highgui.pyc
kalman.py
kalman.pyc
matlab_syntax.py
matlab_syntax.pyc
_cv.pyd
_highgui.pyd
__init__.py
__init__.pyc
其中最重要的是红色的文件
3.写一个测试程序
这个程序用来做autolighting的repeatibility测试。主要原理是:遍历目录下所有的bmp图像,对这些图像的固定一ROI内计算average和SD,并将统计结果生成html文件
1import copy
2import os,sys
3import opencv, highgui
4
5class ROI:
6 def __init__(self, x,y,w,h):
7 self.x = x
8 self.y = y
9 self.w = w
10 self.h = h
11
12def ComputeSDAndAvg(fileName, x,y,w,h):
13 img = opencv.highgui.cvLoadImage(fileName, 0)
14 sumImg = opencv.cv.cvGetSubRect(img, opencv.cv.cvRect(x,y,w,h) )
15 avg = opencv.cv.cvAvg(sumImg)
16 mean = opencv.cv.cvScalar(0)
17 std_dev = opencv.cv.cvScalar(0)
18 opencv.cv.cvAvgSdv(sumImg, mean, std_dev)
19
20 #Modify image
21 img = opencv.highgui.cvLoadImage(fileName, 1)
22 opencv.cv.cvRectangle(img, opencv.cv.cvPoint(x,y),opencv.cv.cvPoint(x+w,y+h),opencv.cv.cvScalar(0,0,255))
23 fileNameOut = fileName + '.jpg'
24 opencv.highgui.cvSaveImage(fileNameOut, img)
25 #return [fileNameOut, avg[0] ]
26 return [fileNameOut, avg[0] , mean[0], std_dev[0]] # Note: avg == mean
27
28def PrintHtml(result,fileName):
29 fp = open(fileName, "w")
30
31 fp.write( '<font color=red>===Image compare tool report===</font><br>\n' )
32 #compute Min-Max avg
33 a=copy.deepcopy(result)
34 a.sort(lambda x,y: int(x[1]-y[1]))
35 fp.write( 'avg MIN %f MAX %f DIFF %f <br>\n' %(a[0][1], a[-1][1], -a[0][1] + a[-1][1]) )
36
37 #print table
38 fp.write( '<table border=1>\n' )
39 fp.write( '<tr><td>Image</td><td>Avg</td><td>SD</td></tr>\n' )
40 for res in result:
41 fp.write( '<tr><td> <img width=400 height=400 src=%s > </td><td>%f,%f</td><td>%f</td></tr>\n' %(res[0], res[1], res[2], res[3]) )
42 fp.write( '</table>\n' )
43 fp.close()
44
45top = r'D:\Python26\Lib\site-packages\opencv'
46result = []
47xywh=[1,1,10,10]
48if len(sys.argv)==5:
49 xywh = [ int(sys.argv[1]), int(sys.argv[2]), int(sys.argv[3]), int(sys.argv[4])]
50else:
51 print "Using default ROI"
52
53for root, dirs, files in os.walk(top, topdown=True):
54
55 for name in files:
56 if name.lower().endswith('.bmp'):
57 fullPath = os.path.join(root, name)
58 if len(fullPath)>20:
59 print fullPath[:10] + '' +fullPath[-10:]
60 else:
61 print fullPath
62 [fileNameOut, avg, mean, std_dev ]= ComputeSDAndAvg(fullPath, xywh[0], xywh[1], xywh[2], xywh[3])
63 result.append([fileNameOut, avg, mean, std_dev ])
64
65 for name in dirs:
66 #os.rmdir(os.path.join(root, name))
67 pass
68htmlFile = os.path.join(top, 'result.html')
69print htmlFile
70PrintHtml(result, htmlFile)
71os.system('explorer.exe "%s"' %htmlFile)
2import os,sys
3import opencv, highgui
4
5class ROI:
6 def __init__(self, x,y,w,h):
7 self.x = x
8 self.y = y
9 self.w = w
10 self.h = h
11
12def ComputeSDAndAvg(fileName, x,y,w,h):
13 img = opencv.highgui.cvLoadImage(fileName, 0)
14 sumImg = opencv.cv.cvGetSubRect(img, opencv.cv.cvRect(x,y,w,h) )
15 avg = opencv.cv.cvAvg(sumImg)
16 mean = opencv.cv.cvScalar(0)
17 std_dev = opencv.cv.cvScalar(0)
18 opencv.cv.cvAvgSdv(sumImg, mean, std_dev)
19
20 #Modify image
21 img = opencv.highgui.cvLoadImage(fileName, 1)
22 opencv.cv.cvRectangle(img, opencv.cv.cvPoint(x,y),opencv.cv.cvPoint(x+w,y+h),opencv.cv.cvScalar(0,0,255))
23 fileNameOut = fileName + '.jpg'
24 opencv.highgui.cvSaveImage(fileNameOut, img)
25 #return [fileNameOut, avg[0] ]
26 return [fileNameOut, avg[0] , mean[0], std_dev[0]] # Note: avg == mean
27
28def PrintHtml(result,fileName):
29 fp = open(fileName, "w")
30
31 fp.write( '<font color=red>===Image compare tool report===</font><br>\n' )
32 #compute Min-Max avg
33 a=copy.deepcopy(result)
34 a.sort(lambda x,y: int(x[1]-y[1]))
35 fp.write( 'avg MIN %f MAX %f DIFF %f <br>\n' %(a[0][1], a[-1][1], -a[0][1] + a[-1][1]) )
36
37 #print table
38 fp.write( '<table border=1>\n' )
39 fp.write( '<tr><td>Image</td><td>Avg</td><td>SD</td></tr>\n' )
40 for res in result:
41 fp.write( '<tr><td> <img width=400 height=400 src=%s > </td><td>%f,%f</td><td>%f</td></tr>\n' %(res[0], res[1], res[2], res[3]) )
42 fp.write( '</table>\n' )
43 fp.close()
44
45top = r'D:\Python26\Lib\site-packages\opencv'
46result = []
47xywh=[1,1,10,10]
48if len(sys.argv)==5:
49 xywh = [ int(sys.argv[1]), int(sys.argv[2]), int(sys.argv[3]), int(sys.argv[4])]
50else:
51 print "Using default ROI"
52
53for root, dirs, files in os.walk(top, topdown=True):
54
55 for name in files:
56 if name.lower().endswith('.bmp'):
57 fullPath = os.path.join(root, name)
58 if len(fullPath)>20:
59 print fullPath[:10] + '' +fullPath[-10:]
60 else:
61 print fullPath
62 [fileNameOut, avg, mean, std_dev ]= ComputeSDAndAvg(fullPath, xywh[0], xywh[1], xywh[2], xywh[3])
63 result.append([fileNameOut, avg, mean, std_dev ])
64
65 for name in dirs:
66 #os.rmdir(os.path.join(root, name))
67 pass
68htmlFile = os.path.join(top, 'result.html')
69print htmlFile
70PrintHtml(result, htmlFile)
71os.system('explorer.exe "%s"' %htmlFile)