Support dynamic batch_size based on sources (#6)

Signed-off-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com>

Author: RizwanMunawar

streamgrid/grid.py

@@ -11,7 +11,7 @@ class StreamGrid:
 
     def __init__(self, sources, model=None, batch_size=4):
         self.sources = sources
-        self.max_sources = len(sources)
+        self.max_sources = self.batch_size = len(sources)
         self.cols = int(math.ceil(math.sqrt(self.max_sources)))
         self.rows = int(math.ceil(self.max_sources / self.cols))
 
@@ -32,7 +32,6 @@ def __init__(self, sources, model=None, batch_size=4):
 
         # Batch processing
         self.model = model
-        self.batch_size = batch_size
         self.frame_queue = queue.Queue(maxsize=50)
 
         # Colors
@@ -222,4 +221,4 @@ def update_display(self):
 
     def stop(self):
         """Stop display."""
-        self.running = False
+        self.running = False