Update app.py

app.py

@@ -119,122 +119,126 @@ RedLine = [173, 0, 173, 600]
 totalCountUp = []
 #mask=cv2.imread('mask.jpg')
 entry_count = 0
-while True:
+
+try:
+    while True:
+        
+        
     
+        success, img = cap.read()
+        # imgRegion=cv2.bitwise_and(img,mask)
     
-
-    success, img = cap.read()
-    # imgRegion=cv2.bitwise_and(img,mask)
-
-    results = model(img, stream=True)
-
-    detections = np.empty((0, 5))
-
-    for r in results:
-        boxes = r.boxes
-        for box in boxes:
-            # Bounding Box
-            x1, y1, x2, y2 = box.xyxy[0]
+        results = model(img, stream=True)
+    
+        detections = np.empty((0, 5))
+    
+        for r in results:
+            boxes = r.boxes
+            for box in boxes:
+                # Bounding Box
+                x1, y1, x2, y2 = box.xyxy[0]
+                x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
+                # cv2.rectangle(img,(x1,y1),(x2,y2),(255,0,255),3)
+                w, h = x2 - x1, y2 - y1
+    
+                # Confidence
+                conf = math.ceil((box.conf[0] * 100)) / 100
+                # Class Name
+                cls = int(box.cls[0])
+                currentClass = classNames[cls]
+    
+                if currentClass == "person" and conf > 0.3:
+                    # cvzone.putTextRect(img, f'{currentClass} {conf}', (max(0, x1), max(35, y1)),
+                    #                    scale=0.6, thickness=1, offset=3)
+                    # cvzone.cornerRect(img, (x1, y1, w, h), l=9, rt=5)
+                    currentArray = np.array([x1, y1, x2, y2, conf])
+                    detections = np.vstack((detections, currentArray))
+    
+        resultsTracker = tracker.update(detections)
+    
+        cv2.line(img, (yelloLine[0], yelloLine[1]), (yelloLine[2], yelloLine[3]), (0, 0, 255), 5)
+        cv2.line(img, (RedLine[0], RedLine[1]), (RedLine[2], RedLine[3]), (0, 255, 200), 5)
+        for result in resultsTracker:
+            x1, y1, x2, y2, id = result
             x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
-            # cv2.rectangle(img,(x1,y1),(x2,y2),(255,0,255),3)
+            print(result)
             w, h = x2 - x1, y2 - y1
-
-            # Confidence
-            conf = math.ceil((box.conf[0] * 100)) / 100
-            # Class Name
-            cls = int(box.cls[0])
-            currentClass = classNames[cls]
-
-            if currentClass == "person" and conf > 0.3:
-                # cvzone.putTextRect(img, f'{currentClass} {conf}', (max(0, x1), max(35, y1)),
-                #                    scale=0.6, thickness=1, offset=3)
-                # cvzone.cornerRect(img, (x1, y1, w, h), l=9, rt=5)
-                currentArray = np.array([x1, y1, x2, y2, conf])
-                detections = np.vstack((detections, currentArray))
-
-    resultsTracker = tracker.update(detections)
-
-    cv2.line(img, (yelloLine[0], yelloLine[1]), (yelloLine[2], yelloLine[3]), (0, 0, 255), 5)
-    cv2.line(img, (RedLine[0], RedLine[1]), (RedLine[2], RedLine[3]), (0, 255, 200), 5)
-    for result in resultsTracker:
-        x1, y1, x2, y2, id = result
-        x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
-        print(result)
-        w, h = x2 - x1, y2 - y1
-        cvzone.cornerRect(img, (x1, y1, w, h), l=9, rt=2, colorR=(255, 0, 255))
-        cvzone.putTextRect(img, f' {int(id)}', (max(0, x1), max(35, y1)),
-                           scale=2, thickness=3, offset=10)
-
-        cx, cy = x1 + w // 2, y1 + h // 2
-        cv2.circle(img, (cx, cy), 5, (255, 0, 255), cv2.FILLED)
-
-        if yelloLine[0] - 20 < cx < yelloLine[2] + 20:
-            if totalCountUp.count(id) == 0:
-                totalCountUp.append(id)
-                dict[id] = [False]
-                cv2.line(img, (yelloLine[0], yelloLine[1]), (yelloLine[2], yelloLine[3]), (0, 0, 255), 5)
-            elif totalCountUp.count(id) == 1:
-                if (dict[id].count(False) < 1):
-                    dict[id].append(False)
-                cv2.line(img, (yelloLine[0], yelloLine[1]), (yelloLine[2], yelloLine[3]), (0, 0, 255), 5)
-        if RedLine[0] - 20 < cx < RedLine[2] + 30:
-            if totalCountUp.count(id) == 0:
-                totalCountUp.append(id)
-                dict[id] = [True]
-                dict[id].append(currentCoordinates)
-                #adding timestamp
-                current_time = datetime.datetime.now()
-                dict[id].append(current_time)
-                cv2.line(img, (RedLine[0], RedLine[1]), (RedLine[2], RedLine[3]), (0, 255, 200), 5)
-            elif totalCountUp.count(id) == 1:
-                if (dict[id].count(True) < 1):
-                    dict[id].append(True)
+            cvzone.cornerRect(img, (x1, y1, w, h), l=9, rt=2, colorR=(255, 0, 255))
+            cvzone.putTextRect(img, f' {int(id)}', (max(0, x1), max(35, y1)),
+                               scale=2, thickness=3, offset=10)
+    
+            cx, cy = x1 + w // 2, y1 + h // 2
+            cv2.circle(img, (cx, cy), 5, (255, 0, 255), cv2.FILLED)
+    
+            if yelloLine[0] - 20 < cx < yelloLine[2] + 20:
+                if totalCountUp.count(id) == 0:
+                    totalCountUp.append(id)
+                    dict[id] = [False]
+                    cv2.line(img, (yelloLine[0], yelloLine[1]), (yelloLine[2], yelloLine[3]), (0, 0, 255), 5)
+                elif totalCountUp.count(id) == 1:
+                    if (dict[id].count(False) < 1):
+                        dict[id].append(False)
+                    cv2.line(img, (yelloLine[0], yelloLine[1]), (yelloLine[2], yelloLine[3]), (0, 0, 255), 5)
+            if RedLine[0] - 20 < cx < RedLine[2] + 30:
+                if totalCountUp.count(id) == 0:
+                    totalCountUp.append(id)
+                    dict[id] = [True]
                     dict[id].append(currentCoordinates)
-                     #adding timestamp
+                    #adding timestamp
                     current_time = datetime.datetime.now()
                     dict[id].append(current_time)
-                    
                     cv2.line(img, (RedLine[0], RedLine[1]), (RedLine[2], RedLine[3]), (0, 255, 200), 5)
-
-    print(totalCountUp)
-    entry_count=0
-
-    for i in dict.values():
-        if (len(i) >= 2):
-            if i[0] == True and i[1] == False:
-                print('in true/False')
-                if entry_count > 0:
-                    entry_count -= 1
-                   
-            if i[0] == False and i[1] == True:
-                print('in /False/True')
-                entry_count += 1
-                
-        
+                elif totalCountUp.count(id) == 1:
+                    if (dict[id].count(True) < 1):
+                        dict[id].append(True)
+                        dict[id].append(currentCoordinates)
+                         #adding timestamp
+                        current_time = datetime.datetime.now()
+                        dict[id].append(current_time)
+                        
+                        cv2.line(img, (RedLine[0], RedLine[1]), (RedLine[2], RedLine[3]), (0, 255, 200), 5)
     
-    # print('count is ', entry_count)
-    # print(dict)
-
+        print(totalCountUp)
+        entry_count=0
     
-    cv2.putText(img, str(entry_count), (110, 245), cv2.FONT_HERSHEY_PLAIN, 5, (50, 50, 230), 7)
-    print('count is ', entry_count)
-    print(dict)
+        for i in dict.values():
+            if (len(i) >= 2):
+                if i[0] == True and i[1] == False:
+                    print('in true/False')
+                    if entry_count > 0:
+                        entry_count -= 1
+                       
+                if i[0] == False and i[1] == True:
+                    print('in /False/True')
+                    entry_count += 1
+                    
+            
+        
+        # print('count is ', entry_count)
+        # print(dict)
     
-    # Create the Gradio interface
-    # interface = gr.Interface(
-    #     fn=returnRBG(img),
-    #     inputs=gr.Text("enter anything"),
-    #     outputs="image",
-    # )
-    # cv2.waitKey(1)
-    print(entry_count)
-    
-    # if currentCoordinates==stationToCoordinates:
-    #     reachedDestination=True
-    #     fareCalculations()
-    #     dict=[]
-    # break
-    if entry_count>2:
-        print('reached destination')
-        fareCalculations()
-        break
+        
+        cv2.putText(img, str(entry_count), (110, 245), cv2.FONT_HERSHEY_PLAIN, 5, (50, 50, 230), 7)
+        print('count is ', entry_count)
+        print(dict)
+        
+        # Create the Gradio interface
+        # interface = gr.Interface(
+        #     fn=returnRBG(img),
+        #     inputs=gr.Text("enter anything"),
+        #     outputs="image",
+        # )
+        # cv2.waitKey(1)
+        print(entry_count)
+        
+        # if currentCoordinates==stationToCoordinates:
+        #     reachedDestination=True
+        #     fareCalculations()
+        #     dict=[]
+        # break
+        if entry_count>2:
+            print('reached destination')
+            fareCalculations()
+            break
+except Exception as e:
+    print("error occured",e)