import sys import cv2 import numpy as np from json import load from os.path import exists, basename from common import image_resize, display, read_json holeConstant = .0156862745 # 160/10200 # use top left, top right, bottom left def apply_image_to_points (original, target, points) : rows, cols, ch = original.shape ir, ic, ich = target.shape print('Using points: ') print(points) atPts = np.float32(points) targetPts = np.float32([[0, 0], [ic, 0], [0, ir]]) M = cv2.getAffineTransform(targetPts, atPts) dst = cv2.warpAffine(target, M, (cols, rows), borderMode=cv2.BORDER_TRANSPARENT) output = original.copy() output[0:rows, 0:cols] = dst return output def create_blank(width, height): rgb_color=(255,255,255) image = np.zeros((height, width, 3), np.uint8) color = tuple(reversed(rgb_color)) image[:] = color gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY) return cv2.cvtColor(gray, cv2.COLOR_GRAY2RGB) # get points 0, 1 and 3 = top left, top right and bottom left def to_points (d) : return ((d['0']['x'], d['0']['y']),(d['1']['x'], d['1']['y']),(d['3']['x'], d['3']['y']),) if len(sys.argv) < 2 : print('Please provide an output destination file') exit(1) outputFile = sys.argv[1] if len(sys.argv) < 3: print('Please provide a calibration template to apply images to') exit(2) templateFile = sys.argv[2] if not exists(templateFile) : print('Calibration template does not exist, please provide one that does') exit(3) if len(sys.argv) < 4: print('Please provide at least one image to apply') exit(4) if len(sys.argv) > 7: print('Please provide maximum four images to apply') exit(5) images = [] for i in range(3, len(sys.argv)): imagePath = sys.argv[i] if not exists(imagePath) : print(f'Image {imagePath} does not exist, exiting...') exit(6) images.append(imagePath) print('Using images: ') for img in images: print(f' -> {basename(img)}') tmpl = read_json(templateFile) print(f"Image size {tmpl['width']}x{tmpl['height']}") hole = round(tmpl['width'] * holeConstant) #blank = create_blank(tmpl['width'], tmpl['height']) output = cv2.imread(templateFile.replace('.calibration.json', ''))#blank.copy() hp = tmpl['holePunches'] for i in hp: print(hp[i]) cv2.circle(output, (hp[i]['x'], hp[i]['y'],), hole, (0, 0, 0,), -1) #cv2.imwrite(outputFile, output) for i in range(0, len(images)) : frame = cv2.imread(images[i]) output = apply_image_to_points(output, frame, to_points(tmpl[f'{i}'])) print(f'Applied {basename(images[i])}') cv2.imwrite(outputFile, output) print(f'Wrote {outputFile}')