Update install.py
Fixes library names so launch.is_installed() properly checks for installed packagespull/38/head
parent
b179be6192
commit
c83dd8c53a
10
install.py
10
install.py
|
|
@ -2,12 +2,12 @@ import launch
|
|||
|
||||
packages = {
|
||||
"onnx": "onnx",
|
||||
"onnxruntime-gpu": "onnxruntime-gpu==1.14.0",
|
||||
"opencv-python": "opencv-python",
|
||||
"onnxruntime": "onnxruntime-gpu==1.14.0",
|
||||
"cv2": "opencv-python",
|
||||
"numpy": "numpy",
|
||||
"Pillow": "Pillow",
|
||||
"segmentation-refinement": "segmentation-refinement",
|
||||
"scikit-learn": "scikit-learn",
|
||||
"PIL": "Pillow",
|
||||
"segmentation_refinement": "segmentation-refinement",
|
||||
"sklearn": "scikit-learn",
|
||||
"clip": "clip",
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue