From 91649d41e53aeb9568b99c72d6e51358eed32c15 Mon Sep 17 00:00:00 2001
From: udon-universe <128375799+udon-universe@users.noreply.github.com>
Date: Sun, 21 May 2023 17:52:52 +0900
Subject: [PATCH 1/2] update package list with dict and pin

---
 install.py | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/install.py b/install.py
index bc238a2..f8bd7b8 100644
--- a/install.py
+++ b/install.py
@@ -1,19 +1,19 @@
 import launch
 
-packages = [
-    "onnx",
-    "onnxruntime-gpu",
-    "opencv-python",
-    "numpy",
-    "Pillow",
-    "segmentation-refinement",
-    "scikit-learn",
-    "clip",
-]
+packages = {
+    "onnx": "onnx",
+    "onnxruntime-gpu": "onnxruntime-gpu==1.14.0",
+    "opencv-python": "opencv-python",
+    "numpy": "numpy",
+    "Pillow": "numpy",
+    "segmentation-refinement": "segmentation-refinement",
+    "scikit-learn": "scikit-learn",
+    "clip": "clip",
+}
 
-for package in packages:
-    if not launch.is_installed(package):
-        launch.run_pip(f'install {package}', desc=f'{package} for PBRemTools')
+for name, target in packages.items():
+    if not launch.is_installed(name):
+        launch.run_pip(f'install {target}', desc=f'{name} for PBRemTools')
 
 if not launch.is_installed("segment_anything"):
     launch.run_pip("install git+https://github.com/facebookresearch/segment-anything.git", desc="segment_anything for PBRemTools")

From efe67520046aa5a90739c842d7dcbb66e66605dd Mon Sep 17 00:00:00 2001
From: udon-universe <128375799+udon-universe@users.noreply.github.com>
Date: Sun, 21 May 2023 20:18:06 +0900
Subject: [PATCH 2/2] fix typo

---
 install.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/install.py b/install.py
index f8bd7b8..2f7bd93 100644
--- a/install.py
+++ b/install.py
@@ -5,7 +5,7 @@ packages = {
     "onnxruntime-gpu": "onnxruntime-gpu==1.14.0",
     "opencv-python": "opencv-python",
     "numpy": "numpy",
-    "Pillow": "numpy",
+    "Pillow": "Pillow",
     "segmentation-refinement": "segmentation-refinement",
     "scikit-learn": "scikit-learn",
     "clip": "clip",