def main(): image0 = cv.LoadImage('tests/grid.jpg') image1 = cv.LoadImage('tests/grid2.jpg') buff = cv.LoadImage('buff.jpg') buff2 = cv.LoadImage('buff.jpg') #image[ y, x , rgb ] features0 = [] for row in csv.reader(open('tests/grid.csv')): features0.append([float(r) for r in row]) features0 = numpy.mat(features0) features1 = [] for row in csv.reader(open('tests/grid2.csv')): features1.append([float(r) for r in row]) features1 = numpy.mat(features1) fund = fundamental(features0, features1) H1, H2 = H1H2Calc(fund) prewarp1, prewarp2 = WarpImages(image0, image1, H1, H2, buff, buff2) ## features0, features1 = formatForFund(features0, features1) ## warpFeatures0 = WarpFeatures(features0.T, H1) ## warpFeatures1 = WarpFeatures(features1.T, H2) #Transition(prewarp1, prewarp2, warpFeatures0, warpFeatures1, features0, features1) cv.NamedWindow('display') ## cv.NamedWindow('prewarp1') ## cv.NamedWindow('prewarp2') writer = cv.CreateVideoWriter("feed/out.avi", 0, 10, (1000,1000), True) ## cv.ShowImage('prewarp1', prewarp1) ## cv.WaitKey(0) ## cv.ShowImage('prewarp2', prewarp2) ## cv.WaitKey(0) Test(prewarp1, prewarp2, H1, H2) Linear(prewarp1, prewarp2, H1, H2, writer)
def main(): image0 = cv.LoadImage('pic1.jpg') image1 = cv.LoadImage('pic2.jpg') buff = cv.LoadImage('buff.jpg') buff2 = cv.LoadImage('buff.jpg') #image[ y, x , rgb ] features0 = numpy.mat([[1771,1111],[2073.5,1056],[1963.5,1259.5],[1732.5,1435.5],[2095.5,1347.5], [1908.5,1468.5],[1941.5,1666.5],[1210,1705],[2156,1551],[1534.5,2040.5], [1952.5,1941.5],[1837,418],[1930.5,1100],[1611.5,1133],[2194.5,1039.5], [1848,797.5],[2101,775.5],[1545.5,1408],[2167,1303.5]]) features1 = numpy.mat([[1738,1111],[2117.5,1094.5],[1936,1309],[1710.5,1457.5],[2161.5,1430], [1919.5,1512.5],[1925,1732.5],[1342,1633.5],[2420,1650],[1644.5,2029.5], [2128.5,2035],[1941.5,374],[1936,1122],[1578.5,1111],[2288,1089],[1798.5,786.5], [2095.5,803],[1540,1391.5],[2293.5,1364]]) fund = fundamental(features0, features1) H1, H2 = H1H2Calc(fund) prewarp1, prewarp2 = WarpImages(image0, image1, H1, H2, buff, buff2) features0, features1 = formatForFund(features0, features1) warpFeatures0 = WarpFeatures(features0.T, H1) warpFeatures1 = WarpFeatures(features1.T, H2) #Transition(prewarp1, prewarp2, warpFeatures0, warpFeatures1, features0, features1) cv.NamedWindow('display') ## cv.NamedWindow('prewarp1') ## cv.NamedWindow('prewarp2') ## cv.ShowImage('prewarp1', prewarp1) ## cv.WaitKey(0) ## cv.ShowImage('prewarp2', prewarp2) ## cv.WaitKey(0) Linear(prewarp1, prewarp2, H1, H2, buff)#, writer)
## e1 = V[:,0] #### print V ## print e1 Fd0 = fundamental * d0 ## print Fd0 Fd0[2] = Fd0[2]**2 Fd0 = Fd0 / math.sqrt(Fd0.sum()) ## print Fd0 a = -Fd0[1] b = Fd0[0] c = numpy.mat("0") d1 = numpy.bmat('a;b;c') ## print d1 return e0, d0, e1, d1 if __name__ == "__main__": fund = fundamental(numpy.mat([[1771,1111],[2073.5,1056],[1963.5,1259.5],[1732.5,1435.5],[2095.5,1347.5], [1908.5,1468.5],[1941.5,1666.5],[1210,1705],[2156,1551],[1534.5,2040.5], [1952.5,1941.5],[1837,418],[1930.5,1100],[1611.5,1133],[2194.5,1039.5], [1848,797.5],[2101,775.5],[1545.5,1408],[2167,1303.5]]), numpy.mat([[1738,1111],[2117.5,1094.5],[1936,1309],[1710.5,1457.5],[2161.5,1430], [1919.5,1512.5],[1925,1732.5],[1342,1633.5],[2420,1650],[1644.5,2029.5], [2128.5,2035],[1941.5,374],[1936,1122],[1578.5,1111],[2288,1089],[1798.5,786.5] ,[2095.5,803],[1540,1391.5],[2293.5,1364]])) H1H2Calc(fund)