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

Merge pull request #24 from udon-universe/feature/fix-duplicate-install-dialog-and-refactor 

fix duplicate install dialog and refactor
pull/30/head
mattya_monaca 2023-05-09 18:51:32 +09:00 committed by GitHub
parent 6abc712c0a 7ef420e15c
commit 124e8a850d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
install.py
View File

@ -1,28 +1,19 @@
import launch
if not launch.is_installed("onnx"):
launch.run_pip("install onnx", "Installing onnx...")
packages = [
"onnx",
"onnxruntime-gpu",
"opencv-python",
"numpy",
"Pillow",
"segmentation-refinement",
"scikit-learn",
"clip",
]
if not launch.is_installed("onnxruntime-gpu"):
launch.run_pip("install onnxruntime-gpu", "Installing onnxruntime-gpu...")
if not launch.is_installed("opencv-python"):
launch.run_pip("install opencv-python", "Installing opencv-python...")
if not launch.is_installed("numpy"):
launch.run_pip("install numpy", "Installing numpy...")
if not launch.is_installed("Pillow"):
launch.run_pip("install Pillow", "Installing Pillow...")
if not launch.is_installed("segmentation-refinement"):
launch.run_pip("install segmentation-refinement", "Installing segmentation-refinement...")
if not launch.is_installed("scikit-learn"):
launch.run_pip("install scikit-learn", "Installing scikit-learn...")
if not launch.is_installed("clip"):
launch.run_pip("install clip", "Installing clip...")
for package in packages:
if not launch.is_installed(package):
launch.run_pip(f'install {package}', desc=f'{package} for PBRemTools')
if not launch.is_installed("segment_anything"):
launch.run_pip("install git+https://github.com/facebookresearch/segment-anything.git", "Installing segment_anything...")
launch.run_pip("install git+https://github.com/facebookresearch/segment-anything.git", desc="segment_anything for PBRemTools")