72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
import sys
|
|
from pathlib import Path
|
|
from textwrap import dedent
|
|
|
|
from packaging import version
|
|
|
|
import launch
|
|
from launch import is_installed, run, run_pip
|
|
|
|
try:
|
|
skip_install = launch.args.skip_install
|
|
except Exception:
|
|
skip_install = False
|
|
|
|
python = sys.executable
|
|
|
|
def check_ddetailer() -> bool:
|
|
try:
|
|
from modules.paths import extensions_dir
|
|
|
|
extensions_path = Path(extensions_dir)
|
|
except ImportError:
|
|
from modules.paths import data_path
|
|
|
|
extensions_path = Path(data_path, "extensions")
|
|
|
|
ddetailer_exists = any(p.is_dir() and p.name.startswith("ddetailer") for p in extensions_path.iterdir())
|
|
return not ddetailer_exists
|
|
|
|
|
|
def check_install() -> bool:
|
|
try:
|
|
import mmcv
|
|
import mmdet
|
|
from mmdet.evaluation import get_classes
|
|
except Exception:
|
|
return False
|
|
|
|
if not hasattr(mmcv, "__version__") or not hasattr(mmdet, "__version__"):
|
|
return False
|
|
|
|
v1 = version.parse(mmcv.__version__) >= version.parse("2.0.0")
|
|
v2 = version.parse(mmdet.__version__) >= version.parse("3.0.0")
|
|
return v1 and v2
|
|
|
|
|
|
def install():
|
|
if not is_installed("pycocotools"):
|
|
run(f"{python} -m pip install pycocotools", live=True)
|
|
|
|
if not is_installed("mim"):
|
|
run_pip("install openmim", desc="openmim")
|
|
|
|
if not check_install():
|
|
print("Uninstalling mmcv mmdet... (if installed)")
|
|
run(f'"{python}" -m pip uninstall -y mmcv mmcv-full mmdet mmengine', live=True)
|
|
print("Installing mmcv mmdet...")
|
|
run(f'"{python}" -m mim install -U mmcv>=2.0.0 mmdet>=3.0.0', live=True)
|
|
|
|
|
|
if not check_ddetailer():
|
|
message = """
|
|
[-] dddetailer: Please remove the following:
|
|
1. the original ddetailer extension - "stable-diffusion-webui/extensions/ddetailer" folder.
|
|
2. original model files - "stable-diffusion-webui/models/mmdet" folder.
|
|
"""
|
|
message = dedent(message)
|
|
raise RuntimeError(message)
|
|
|
|
if not skip_install:
|
|
install()
|