Update main.py
Browse files
main.py
CHANGED
|
@@ -35,7 +35,7 @@ ws = wb.active
|
|
| 35 |
ws.append(["Timestamp", "Predicted Image Path", "Label"]) # Headers
|
| 36 |
|
| 37 |
# Initialize video writer
|
| 38 |
-
video_path = "./
|
| 39 |
fourcc = cv2.VideoWriter_fourcc(*"mp4v")
|
| 40 |
fps = 0.5 # Adjust frames per second as needed
|
| 41 |
video_writer = None
|
|
@@ -126,4 +126,4 @@ os.rmdir(screenshots_path)
|
|
| 126 |
print(f"Results saved to {excel_file}")
|
| 127 |
|
| 128 |
# Close OpenCV window
|
| 129 |
-
cv2.destroyAllWindows()
|
|
|
|
| 35 |
ws.append(["Timestamp", "Predicted Image Path", "Label"]) # Headers
|
| 36 |
|
| 37 |
# Initialize video writer
|
| 38 |
+
video_path = "./video/annotated_video.mp4"
|
| 39 |
fourcc = cv2.VideoWriter_fourcc(*"mp4v")
|
| 40 |
fps = 0.5 # Adjust frames per second as needed
|
| 41 |
video_writer = None
|
|
|
|
| 126 |
print(f"Results saved to {excel_file}")
|
| 127 |
|
| 128 |
# Close OpenCV window
|
| 129 |
+
cv2.destroyAllWindows()
|