91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
import numpy as np
|
|
import cv2
|
|
import imutils
|
|
|
|
|
|
class Layer:
|
|
# bounds = [[(x,y,w,h), ],]
|
|
|
|
startFrame = None
|
|
lastFrame = None
|
|
length = None
|
|
|
|
def __init__(self, startFrame, data, mask, config):
|
|
"""returns a Layer object
|
|
|
|
Layers are collections of contours with a StartFrame,
|
|
which is the number of the frame the first contour of
|
|
this layer was extraced from
|
|
|
|
A Contour is a CV2 Contour, which is a y*x*3 rgb numpy array,
|
|
but we only care about the corners of the contours.
|
|
So we save the bounds (x,y,w,h) in bounds[] and the actual content in data[]
|
|
"""
|
|
self.startFrame = startFrame
|
|
self.lastFrame = startFrame
|
|
self.config = config
|
|
self.data = []
|
|
self.bounds = []
|
|
self.masks = []
|
|
self.stats = dict()
|
|
self.exportOffset = 0
|
|
|
|
self.bounds.append([data])
|
|
self.masks.append([mask])
|
|
|
|
def add(self, frameNumber, bound, mask):
|
|
"""Adds a bound to the Layer at the layer index which corresponds to the given framenumber"""
|
|
index = frameNumber - self.startFrame
|
|
if index < 0:
|
|
return
|
|
if frameNumber > self.lastFrame:
|
|
for i in range(frameNumber - self.lastFrame):
|
|
self.bounds.append([])
|
|
self.masks.append([])
|
|
self.lastFrame = frameNumber
|
|
|
|
if bound not in self.bounds[index]:
|
|
self.bounds[index].append(bound)
|
|
self.masks[index].append(mask)
|
|
|
|
def getLength(self):
|
|
return len(self) + self.exportOffset
|
|
|
|
def __len__(self):
|
|
self.length = len(self.bounds)
|
|
return self.length
|
|
|
|
def spaceOverlaps(self, layer2):
|
|
"""Checks if there is an overlap in the bounds of current layer with given layer"""
|
|
overlap = False
|
|
maxLen = min(len(layer2.bounds), len(self.bounds))
|
|
bounds = self.bounds[:maxLen]
|
|
for b1s, b2s in zip(bounds[::10], layer2.bounds[:maxLen:10]):
|
|
for b1 in b1s:
|
|
for b2 in b2s:
|
|
if self.contoursOverlay((b1[0], b1[1] + b1[3]), (b1[0] + b1[2], b1[1]), (b2[0], b2[1] + b2[3]), (b2[0] + b2[2], b2[1])):
|
|
overlap = True
|
|
break
|
|
return overlap
|
|
|
|
def timeOverlaps(self, layer2):
|
|
"""Checks for overlap in time between current and given layer"""
|
|
s1 = self.exportOffset
|
|
e1 = self.lastFrame - self.startFrame + self.exportOffset
|
|
s2 = layer2.exportOffset
|
|
e2 = layer2.lastFrame - layer2.startFrame + layer2.exportOffset
|
|
|
|
if s2 >= s1 and s2 <= e1:
|
|
return True
|
|
elif s1 >= s2 and s1 <= e2:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def contoursOverlay(self, l1, r1, l2, r2):
|
|
if l1[0] >= r2[0] or l2[0] >= r1[0]:
|
|
return False
|
|
if l1[1] <= r2[1] or l2[1] <= r1[1]:
|
|
return False
|
|
return True
|