1580158
/
PBRemTools
mirror of https://github.com/mattyamonaca/PBRemTools.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update package list with dict and pin

pull/34/head
udon-universe 2023-05-21 17:52:52 +09:00
parent dcd00660df
commit 91649d41e5
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
install.py
View File

@ -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")