opencv+python 实现基本图形的绘制及文本的添加

   使用opencv进行一些简单的图形绘制，期间查询借鉴了他人的方法，在python2.7环境下亲测可以使用，可以实现直线，矩形，圆形，椭圆，多边形以及标注文字的添加，设置了一些默认参数，使用更加便捷，使用前需要安装opencv，numpy包，可以直接使用pip install XXX 安装

import cv2
import numpy as np
import os

class Drawing(object):
    """
    使用opencv绘制图形，支持直线，矩形，圆形，椭圆，多边形以及被标注文字添加
    """
    chart_list = ['line', 'rectangle', 'circle', 'ellipse', 'polylines', 'puttext']

    def __init__(self, src_img, dst_img, chart, dict_args):
        self.src_img = os.path.normpath(src_img)
        self.dst_img = os.path.normpath(dst_img)
        self.chart = chart
        self.dict_args = dict_args
        # 颜色不传默认为红色
        self.color = dict_args['color'] if dict_args.has_key('color') else (0,0,255)
        # 线条粗细不传默认为 2
        self.thickness = dict_args['thickness'] if dict_args.has_key('thickness') else 2

    def handle(self):
        # 导入图片
        self.src_img = cv2.imread(self.src_img)
        if self.chart not in self.chart_list:
            print 'must input your right parameter'
            return
        if self.chart == 'line':
            # 画直线
            self.start = self.dict_args['start']
            self.end = self.dict_args['end']
            self.draw_line()
        elif self.chart == 'rectangle':
            # 画矩形
            self.top_left = self.dict_args['top_left']
            self.bottom_right = self.dict_args['bottom_right']
            self.draw_rectangle()
        elif self.chart == 'circle':
            # 画圆形
            self.center = self.dict_args['center']
            self.radius = self.dict_args['radius']
            self.draw_circle()
        elif self.chart == 'ellipse':
            # 画椭圆
            self.center = self.dict_args['center']
            self.axes = self.dict_args['axes']
            # 旋转角度，起始角度，终止角度 可不传参，使用默认值
            self.angle = self.dict_args['angle'] if self.dict_args.has_key('angle') else 0
            self.startangle = self.dict_args['startangle'] if self.dict_args.has_key('startangle') else 0
            self.endangle = self.dict_args['endangle'] if self.dict_args.has_key('endangle') else 360
            self.draw_ellipse()
        elif self.chart == 'polylines':
            # 画多边形
            if not isinstance(self.dict_args['points'], list):
                self.pts = list(self.dict_args['points'])
            self.pts = np.array(self.dict_args['points'], np.int32)
            self.close = self.dict_args['close'] if self.dict_args.has_key('close') else True
            self.draw_polylines()
        else:
            # 标注文本
            self.text = self.dict_args['text']
            self.position = self.dict_args['position']
            # 字体，文字大小 可不传参，使用默认值
            self.font = self.dict_args['font'] if self.dict_args.has_key('font') else cv2.FONT_HERSHEY_SIMPLEX
            self.size = self.dict_args['size'] if self.dict_args.has_key('size') else 1
            self.add_text()
        cv2.imwrite(self.dst_img, self.src_img)

    def draw_line(self):
        # 划线
        # 输入参数分别为图像，开始坐标，结束坐标，颜色数组，粗细
        cv2.line(self.src_img, self.start, self.end, self.color, self.thickness)

    def draw_rectangle(self):
        # 画矩形
        # 输入参数分别为图像、左上角坐标、右下角坐标、颜色数组、粗细
        cv2.rectangle(self.src_img, self.top_left, self.bottom_right, self.color, self.thickness)

    def draw_circle(self):
        # 画圆形
        # 输入参数为图像，圆心，半径，线条颜色，粗细
        cv2.circle(self.src_img, self.center, self.radius, self.color, self.thickness)

    def draw_ellipse(self):
        # 画椭圆
        # 输入参数为图像，中心，（长轴，短轴），旋转角度，起始角度，终止角度，线条颜色，粗细
        cv2.ellipse(self.src_img, self.center, self.axes, self.angle, self.startangle,self.endangle, self.color, self.thickness)

    def draw_polylines(self):
        # 画多边形
        # 输入参数为图像，多边形各个顶点坐标，是否连成封闭图形，线的颜色，粗细
        cv2.polylines(self.src_img, [self.pts], self.close, self.color, self.thickness)

    def add_text(self):
        # 标注文本
        # 输入参数为图像、文本、位置、字体、大小、颜色数组、粗细
        cv2.putText(self.src_img, self.text, self.position, self.font, self.size, self.color, self.thickness)