samples/python/stitching.py

一个关于Python图像拼接的基本示例。

#!/usr/bin/env python
 
'''
图像拼接示例
================
 
演示如何以简单的方式使用Python中的Stitcher API进行全景图或扫描图像的拼接。
。
'''
 
# Python 2/3兼容性
from __future__ import print_function
 
import numpy as np
import cv2 as cv
 
import argparse
import sys
 
modes = (cv.Stitcher_PANORAMA, cv.Stitcher_SCANS)
 
parser = argparse.ArgumentParser(prog='stitching.py', description='图像拼接示例。')
parser.add_argument('--mode',
 type = int, choices = modes, default = cv.Stitcher_PANORAMA,
 help = '确定拼接器的配置。默认值为`PANORAMA` (%d)，'
 '适用于创建照片全景图的模式。选项`SCANS` (%d) 适用于'
 '拼接仿射变换下的材料，例如扫描件。' % modes)
parser.add_argument('--output', default = 'result.jpg',
 help = '结果图像。默认为`result.jpg`。')
parser.add_argument('img', nargs='+', help = '输入图像')
 
__doc__ += '\n' + parser.format_help()
 
def main()
 args = parser.parse_args()
 
 # 读取输入图像
 imgs = []
 for img_name in args.img
 img = cv.imread(cv.samples.findFile(img_name))
 if img is None
 print("无法读取图像 " + img_name)
 sys.exit(-1)
 imgs.append(img)
 
 #![图像拼接]
 stitcher = cv.Stitcher.create(args.mode)
 status, pano = stitcher.stitch(imgs)
 
 if status != cv.Stitcher_OK
 print("无法拼接图像，错误代码 = %d" % status)
 sys.exit(-1)
 #![图像拼接]
 
 cv.imwrite(args.output, pano)
 print("图像拼接完成。已保存 %s!" % args.output)
 
 print('完成')
 
 
if __name__ == '__main__'
 print(__doc__)
 main()
 cv.destroyAllWindows()