Update run.py

run.py

@@ -8,13 +8,16 @@ from transformers import SegformerImageProcessor, SegformerForSemanticSegmentati
 
 
 
-# class Count:
-#     def __init__(self):
-#         self.n = 0
-#         self.imout = np.zeros((1000, 1000))
-        
-#     def step(self):
-#         self.n += 1
+class Counter:
+    def __init__(self):
+        self.count = 0
+
+    def increment(self):
+        self.count += 1
+        return self.count
+
+counter = Counter()
+
 
 
 cnt = 0  
@@ -35,8 +38,8 @@ model.to(device).eval()
 # counter = Count()
 
 def segment(im, interval_s=2):
-    global cnt
-    cnt += 1
+    global counter
+    counter.increment()
     im = cv2.resize(im, (im.shape[1] // 2, im.shape[0] // 2))
     imout = im.copy()