Corrections for YOLO on non-CPU Linux

Author: ChrisMaunder

.vscode/settings.json

@@ -95,6 +95,7 @@
         "logicaldisk",
         "logvals",
         "lproj",
+        "Matplotlib",
         "maxs",
         "megas",
         "megaunit",
@@ -133,6 +134,7 @@
         "postprocess",
         "prerm",
         "pstree",
+        "psutil",
         "pushd",
         "pycoral",
         "pypi",
@@ -175,6 +177,7 @@
         "statuseseses",
         "subshell",
         "Systemmd",
+        "TDQM",
         "Tegra",
         "tegrastats",
         "tempstore",
@@ -188,6 +191,7 @@
         "totalspacebytes",
         "tpus",
         "ufeff",
+        "Ultralytics",
         "unclip",
         "uninstallexe",
         "upsampling",

modules/ObjectDetectionYOLOv5-6.2/requirements.linux.txt

@@ -1,31 +1,42 @@
 #! Python3.7
 
-Pandas                          # Installing Pandas, a data analysis / data manipulation tool
-CoreMLTools                     # Installing CoreMLTools, for working with .mlmodel format models
-OpenCV-Python                   # Installing OpenCV, the Open source Computer Vision library
-Pillow<10.0.0                   # Installing Pillow, a Python Image Library
-SciPy                           # Installing SciPy, a library for mathematics, science, and engineering
-PyYAML                          # Installing PyYAML, a library for reading configuration files
+opencv-python-headless==4.8.1.78    # Installing OpenCV (headless), the Open source Computer Vision library
+numpy==1.26.0                       # Installing NumPy, a package for scientific computing
+matplotlib>=3.3.0                   # Installing Matplotlib, the Python plotting package
+opencv-python>=4.6.0                # Installing OpenCV, the Open source Computer Vision library
+pillow>=7.1.2                       # Installing Pillow, a Python Image Library
+pyyaml>=5.3.1                       # Installing PyYAML, a library for reading configuration files
+requests>=2.23.0                    # Installing Requests, the HTTP library
+scipy>=1.4.1                        # Installing SciPy, a library for mathematics, science, and engineering
+tqdm>=4.64.0                        # Installing TDQM, the Fast, Extensible Progress Meter
+pandas>=1.1.4                       # Installing Pandas, a data analysis / data manipulation tool
+psutil                              # Installing psutil, a tool to check system utilization
+dill                                # Installing dill, for serializing and de-serializing Python objects
+py-cpuinfo                          # Installing py-cpuinfo to allow us to query CPU info
+ipython                             # Installing ipython, for interactive notebooks
 
 # PyTorch-DirectML not working for this module
 # torch-directml                # Installing the PyTorch DirectML plugin
 
---extra-index-url https://download.pytorch.org/whl/cpu
-Torch                           # Installing Torch, for Tensor computation and Deep neural networks
---extra-index-url https://download.pytorch.org/whl/cpu
-TorchVision                     # Installing TorchVision, for Computer Vision based AI
+# CPU specific Torch 1.8.0. Size ~531Mb
+-f https://download.pytorch.org/whl/cpu/torch_stable.html
+torch==1.8.0+cpu                    # Installing PyTorch, for Tensor computation and Deep neural networks
+-f https://download.pytorch.org/whl/cpu/torch_stable.html
+torchvision==0.9.0+cpu              # Installing TorchVision, for Computer Vision based AI
 
-# We'll lock down to Torch 1.X, CPU-only to play it safe
-#--extra-index-url https://download.pytorch.org/whl/cpu
-#torch==1.13.1+cpu             `# Installing Torch, for Tensor computation and Deep neural networks
-#--extra-index-url https://download.pytorch.org/whl/cpu
-#torchvision==0.14.1+cpu       `# Installing TorchVision, for Computer Vision based AI
+# CPU specific Torch 1.13.0. Size ~830Mb
+# --extra-index-url https://download.pytorch.org/whl/cpu
+# torch==1.13.0+cpu                 # Installing PyTorch, for Tensor computation and Deep neural networks
+# --extra-index-url https://download.pytorch.org/whl/cpu
+# torchvision==0.14.0+cpu           # Installing TorchVision, for Computer Vision based AI
 
-yolov5==6.2.3                   # Installing Ultralytics YoloV5 package for object detection in images
+# Explicitly install ultralytics without optional dependencies (like CUDA).
+--no-deps
+yolov5==6.2.3                       # Installing Ultralytics YoloV5 package for object detection in images
 
-# We need this, but we don't need this.
-Seaborn                         # Installing Seaborn, a data visualization library based on matplotlib
+# CoreMLTools                       # Installing CoreMLTools, for working with .mlmodel format models
 
-CodeProject-AI-SDK              # Installing the CodeProject.AI SDK
+
+CodeProject-AI-SDK                  # Installing the CodeProject.AI SDK
 
 # last line empty.

src/demos/modules/PythonSimple/requirements.linux.txt

@@ -1,17 +1,33 @@
 #! Python3.8
 
-# Torch                         # Installing Torch, for Tensor computation and Deep neural networks
-# TorchVision                   # Installing TorchVision, for Computer Vision based AI
+# Based on the answer for pleas to Ultralytics to not include the full bloated NVIDIA libraries
+# when no GPU present: https://github.com/ultralytics/ultralytics/issues/749#issuecomment-1806360075
 
-# We'll lock down to Torch 1.X, CPU-only to play it safe
---extra-index-url https://download.pytorch.org/whl/cpu
-torch==1.13.1+cpu               # Installing Torch, for Tensor computation and Deep neural networks
---extra-index-url https://download.pytorch.org/whl/cpu
-torchvision==0.14.1+cpu         # Installing TorchVision, for Computer Vision based AI
+opencv-python-headless==4.8.1.78    # Installing OpenCV (headless), the Open source Computer Vision library
+numpy==1.26.0                       # Installing NumPy, a package for scientific computing
+matplotlib>=3.3.0                   # Installing Matplotlib, the Python plotting package
+opencv-python>=4.6.0                # Installing OpenCV, the Open source Computer Vision library
+pillow>=7.1.2                       # Installing Pillow, a Python Image Library
+pyyaml>=5.3.1                       # Installing PyYAML, a library for reading configuration files
+requests>=2.23.0                    # Installing Requests, the HTTP library
+scipy>=1.4.1                        # Installing SciPy, a library for mathematics, science, and engineering
+tqdm>=4.64.0                        # Installing TDQM, the Fast, Extensible Progress Meter
+pandas>=1.1.4                       # Installing Pandas, a data analysis / data manipulation tool
+psutil                              # Installing psutil, a tool to check system utilization
+dill                                # Installing dill, for serializing and de-serializing Python objects
+py-cpuinfo                          # Installing py-cpuinfo to allow us to query CPU info
 
-# Specific version because we have a patch
-ultralytics==8.1.2              # Installing Ultralytics package for object detection in images
+#Install torch and torchvision CPU only versions
+-f https://download.pytorch.org/whl/cpu/torch_stable.html
+torch==1.8.0+cpu                    # Installing PyTorch, for Tensor computation and Deep neural networks
+-f https://download.pytorch.org/whl/cpu/torch_stable.html
+torchvision==0.9.0+cpu              # Installing TorchVision, for Computer Vision based AI
 
-CodeProject-AI-SDK              # Installing the CodeProject.AI SDK
+# Explicitly install ultralytics without optional dependencies (like CUDA).
+--no-deps
+ultralytics==8.1.2                  # Installing Ultralytics package for object detection in images
+
+
+CodeProject-AI-SDK                  # Installing the CodeProject.AI SDK
 
 # last line empty.