liangyuch commited on
Commit
581e1e5
β€’
1 Parent(s): 57fb51b

install openpsg

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -16,6 +16,9 @@ if os.getenv('SYSTEM') == 'spaces':
16
  subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
17
  subprocess.call('pip install pycocotools'.split())
18
  subprocess.call("pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html".split())
 
 
 
19
  # pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html
20
  # detectron2==0.6 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
21
  import cv2
 
16
  subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
17
  subprocess.call('pip install pycocotools'.split())
18
  subprocess.call("pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html".split())
19
+ subprocess.call("git clone https://github.com/Jingkang50/OpenPSG.git".split())
20
+ subprocess.call("pip install -e OpenPSG".split())
21
+
22
  # pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html
23
  # detectron2==0.6 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
24
  import cv2