Opencv之生成棋盘标定板

代码展示

# -*- coding:utf-8 -*-
import cv2
import numpy as np


class Chessboard:
    def __init__(self):
        # 背景颜色
        self.background_color = [255, 255, 255]
        # 棋盘颜色
        self.foreground_color = [0, 0, 0]
        # 每个棋盘格占像素个数
        self.block_size_pixel = 8

    def gen_chessboard_image(self, pattern_size_hw):

        height, width = pattern_size_hw
        # 图像大小
        image_height = self.block_size_pixel * (height + 1)
        image_width = self.block_size_pixel * (width + 1)
        # 生成图像
        image = np.zeros((image_height, image_width, 3), dtype=np.uint8)

        image[:] = self.background_color
        if self.block_size_pixel != 0 and 0 not in pattern_size_hw:
            for j in range(height + 1):
                for i in range(width + 1):
                    if (i + j) % 2 != 0:
                        x1 = i * self.block_size_pixel
                        x2 = (i + 1) * self.block_size_pixel
                        y1 = j * self.block_size_pixel
                        y2 = (j + 1) * self.block_size_pixel
                        image[y1: y2, x1: x2, :] = self.foreground_color
        return image


if __name__ == '__main__':
    calibration_board = Chessboard()
    board_pic = calibration_board.gen_chessboard_image(pattern_size_hw=[7, 8])

    cv2.namedWindow('cb', 0)
    cv2.imshow('cb', board_pic)
    cv2.waitKey()

效果展示