Skip to content

Commit

Permalink
Fix: unable to find mmcv._ext build issue
Browse files Browse the repository at this point in the history
  • Loading branch information
dsxailab committed Apr 7, 2023
1 parent 2bcab78 commit f06a7d9
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions scripts/ddetailer.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,11 +45,10 @@ def modeltitle(path, shorthash):

def startup():
from launch import is_installed, run
if not is_installed("mmdet"):
python = sys.executable
run(f'"{python}" -m pip install -U openmim', desc="Installing openmim", errdesc="Couldn't install openmim")
run(f'"{python}" -m mim install mmcv-full', desc=f"Installing mmcv-full", errdesc=f"Couldn't install mmcv-full")
run(f'"{python}" -m pip install mmdet', desc=f"Installing mmdet", errdesc=f"Couldn't install mmdet")
python = sys.executable
run(f'"{python}" -m pip install --upgrade -U openmim', desc="Installing openmim", errdesc="Couldn't install openmim")
run(f'"{python}" -m mim install --upgrade mmcv-full', desc=f"Installing mmcv-full", errdesc=f"Couldn't install mmcv-full")
run(f'"{python}" -m pip install --upgrade mmdet', desc=f"Installing mmdet", errdesc=f"Couldn't install mmdet")

if (len(list_models(dd_models_path)) == 0):
print("No detection models found, downloading...")
Expand Down

0 comments on commit f06a7d9

Please sign in to comment.