51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
import numpy as np
|
|
import cv2
|
|
import imutils
|
|
class Layer:
|
|
#data = [(contour, (x,y,w,h)),]
|
|
|
|
startFrame = None
|
|
lastFrame = None
|
|
length = None
|
|
|
|
def __init__(self, startFrame, data):
|
|
self.startFrame = startFrame
|
|
self.lastFrame = startFrame
|
|
|
|
self.data = []
|
|
self.bounds = []
|
|
self.bounds.append(data)
|
|
|
|
|
|
#print("Layer constructed")
|
|
|
|
def add(self, frameNumber, data):
|
|
if not (self.startFrame + len(self.bounds) - frameNumber < 0):
|
|
self.lastFrame = frameNumber
|
|
|
|
self.bounds.append(data)
|
|
|
|
def getLength(self):
|
|
self.length = len(self.data)
|
|
return self.length
|
|
|
|
def fill(self, inputPath):
|
|
'''reads in the contour data, needed for export'''
|
|
|
|
cap = cv2.VideoCapture(inputPath)
|
|
self.data = [None]*len(self.bounds)
|
|
i = 0
|
|
cap.set(1, self.startFrame)
|
|
while i < len(self.bounds):
|
|
ret, frame = cap.read()
|
|
|
|
if ret:
|
|
frame = imutils.resize(frame, width=512)
|
|
(x, y, w, h) = self.bounds[i]
|
|
self.data[i] = frame[y:y+h, x:x+w]
|
|
i+=1
|
|
cap.release()
|
|
|
|
|
|
|