banished ghosts

Application/Config.py

@@ -1,7 +1,7 @@
 
 class Config:
     c = {
-        "min_area" : 500,
+        "min_area" : 100,
         "max_area" : 40000,
         "threashold" : 8,
         "resizeWidth" : 512,

Application/Layer.py

@@ -39,9 +39,10 @@ class Layer:
                 self.bounds[frameNumber - self.startFrame].append(bound)
                 self.masks[frameNumber - self.startFrame].append(mask)
         else:
+            while len(self.bounds) + self.startFrame < frameNumber:
+                self.bounds.append([bound])
+                self.masks.append([mask])
             self.lastFrame = frameNumber
-            self.bounds.append([bound])
-            self.masks.append([mask])
 
     def calcStats(self):
         '''calculates average distance, variation and deviation of layer movement'''

Application/LayerFactory.py

@@ -41,7 +41,7 @@ class LayerFactory:
                 contours = data[frameNumber]
                 masks = maskArr[frameNumber]
                 masks = [np.unpackbits(mask, axis=0) for mask, contours in zip(masks, contours)]
-                if frameNumber%5000 == 0:
+                if frameNumber%100 == 0:
                     print(f" {int(round(frameNumber/max(data.keys()), 2)*100)}% done with Layer extraction", end='\r')
 
                 tmp = [[frameNumber, contour, mask] for contour, mask in zip(contours, masks)]

generate test footage/gen.py

@@ -41,7 +41,10 @@ def genVideo():
             objectX -= objectSpeedX
             objectY -= objectSpeedY
 
-            objectShape = [(objectX, objectY), (objectX + objectWidth, objectY + objectHeight)] 
+            objectShape = [
+                (objectX, objectY), 
+                (objectX + objectWidth, objectY + objectHeight)
+            ] 
             img = Image.new("RGB", (xmax, ymax)) 
             img1 = ImageDraw.Draw(img)   
             img1.rectangle(objectShape, fill = color) 

main.py

@@ -44,8 +44,8 @@ def main():
         start = time.time()
     else:
         layers, contours, masks = Importer(config).importRawData()
-        #layerFactory = LayerFactory(config)
+        layerFactory = LayerFactory(config)
-        #layers = layerFactory.extractLayers(contours, masks)
+        layers = layerFactory.extractLayers(contours, masks)
 
     layerManager = LayerManager(config, layers)
     layerManager.transformLayers()
@@ -56,7 +56,7 @@ def main():
     layers = layerManager.layers
     exporter = Exporter(config)
     print(f"Exporting {len(contours)} Contours and {len(layers)} Layers")
-    exporter.export(layers, contours, masks, raw=False, overlayed=True)
+    exporter.export(layers, contours, masks, raw=True, overlayed=True)
     stats["Exporter"] = time.time() - start
 
     print("Total time: ", time.time() - startTotal)