"atomic bomb" commit. Reorganized OpenCV directory structure
This commit is contained in:
60
tests/swig_python/highgui/match.py
Executable file
60
tests/swig_python/highgui/match.py
Executable file
@@ -0,0 +1,60 @@
|
||||
"""
|
||||
This script will compare tho images and decides with a threshold
|
||||
if these to images are "equal enough"
|
||||
"""
|
||||
|
||||
# import the necessary things for OpenCV
|
||||
from cv import *
|
||||
from highgui import *
|
||||
|
||||
import frames
|
||||
import sys
|
||||
import os
|
||||
|
||||
PREFIX=os.path.join(os.environ["srcdir"],"../../opencv_extra/testdata/python/images/")
|
||||
|
||||
|
||||
DisplayImages=False
|
||||
|
||||
if DisplayImages:
|
||||
videowindow="video"
|
||||
referencewindow="reference"
|
||||
cvNamedWindow(videowindow,CV_WINDOW_AUTOSIZE)
|
||||
cvNamedWindow(referencewindow,CV_WINDOW_AUTOSIZE)
|
||||
|
||||
# returns True/False if match/non-match
|
||||
def match( image, index, thres ):
|
||||
|
||||
# load image from comparison set
|
||||
QCIFcompare=cvLoadImage(PREFIX+frames.QCIF[index])
|
||||
|
||||
if QCIFcompare is None:
|
||||
print "Couldn't open image "+PREFIX+frames.QCIF[index]+" for comparison!"
|
||||
sys.exit(1)
|
||||
|
||||
# resize comparison image to input image dimensions
|
||||
size=cvSize(image.width,image.height)
|
||||
compare=cvCreateImage(size,IPL_DEPTH_8U,image.nChannels)
|
||||
cvResize(QCIFcompare,compare)
|
||||
|
||||
# compare images
|
||||
diff=cvNorm( image, compare, CV_RELATIVE_L2 )
|
||||
|
||||
if DisplayImages:
|
||||
cvShowImage(videowindow,image)
|
||||
cvShowImage(referencewindow,compare)
|
||||
if diff<=thres:
|
||||
cvWaitKey(200)
|
||||
else:
|
||||
print "index==",index,": max==",thres," is==",diff
|
||||
cvWaitKey(5000)
|
||||
|
||||
cvReleaseImage(QCIFcompare)
|
||||
cvReleaseImage(compare)
|
||||
|
||||
if diff<=thres:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
|
Reference in New Issue
Block a user