fix duplicate log and refactor

pull/24/head
udon-universe 2023-05-02 01:29:56 +09:00
parent 6abc712c0a
commit d768114f79
1 changed files with 14 additions and 23 deletions

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=package)
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")