Updated requirements.txt, added data collection script

Author: sahil-h-patel

data_collection.py

@@ -0,0 +1,53 @@
+###############################################################################
+#
+# File: store_images.py
+#
+# Author: Isaac Ingram
+#
+# Purpose: Store an image stream from a website to the local disk.
+#
+# Usage: This program requires the requests library to be installed.
+# To use, provide two commandline arguments: <URL> and <path>. 'URL' is the URL
+# to requests content from. 'path' is the relative or absolute path to where
+# the images should be placed on your system.
+#
+###############################################################################
+import sys
+from datetime import datetime
+
+import requests
+
+
+def print_usage_and_exit() -> None:
+    """
+    Print the usage statement and exit.
+    :return: None
+    """
+    print("Usage: store_images.py <URL> <path>")
+    exit(0)
+
+
+def main():
+    try:
+        while(True):
+            # Check command line args
+            if len(sys.argv) != 3:
+                print_usage_and_exit()
+            url = sys.argv[1]
+            storage_path = sys.argv[2]
+            # Get rid of last character in path if it's a slash
+            if storage_path[-1] == "/":
+                storage_path = storage_path[:-1]
+            try:
+                image_data = requests.get(url).content
+                img_name = f'{storage_path}/{datetime.now().strftime("%S-%M-%H-%d-%m-%Y.jpg")}'
+                with open(img_name, 'wb') as image_file:
+                    image_file.write(image_data)
+            except Exception as e:
+                print(f"Failed to download from {url}: {e}")
+    except KeyboardInterrupt:
+        print("Program interrupted by user. Exiting...")
+
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -24,7 +24,7 @@ matplotlib==3.10.0
 matplotlib-inline==0.1.7
 mpmath==1.3.0
 networkx==3.4.2
-numpy==1.23.5
+numpy
 opencv-python==4.11.0.86
 packaging==24.2
 paho-mqtt==2.1.0
@@ -54,8 +54,8 @@ sympy==1.13.1
 tensorboard==2.19.0
 tensorboard-data-server==0.7.2
 threadpoolctl==3.6.0
-torch==2.5.1
-torchvision==0.20.1
+torch
+torchvision
 tqdm==4.67.1
 traitlets==5.14.3
 typing_extensions==4.12.2
@@ -64,4 +64,4 @@ ultralytics==8.3.111
 ultralytics-thop==2.0.14
 urllib3==2.3.0
 wcwidth==0.2.13
-Werkzeug==3.1.3
+Werkzeug==3.1.3