背景データを切り出してネガティブデータを作る

ng_maker.py

import math

import cv2
import numpy


count = 0
for i in range(12):
    print('makeing {0}.mp4 background image'.format(i+1))
    video = cv2.VideoCapture('src/{0}.mp4'.format(i+1))

    background = numpy.array(video.read()[1], numpy.uint32)
    frame_count = 1

    while True:
        ok, img = video.read()
        if not ok:
            break

        background += img
        frame_count += 1

    background = numpy.array(background / frame_count, numpy.uint8)

    video.release()

    for x in range(0, background.shape[1]-128, 64):
        for y in range(0, background.shape[0]-128, 64):
            print(count)
            cv2.imwrite('ng/{0}.jpg'.format(count), background[y:y+128, x:x+128])
            count += 1