Qihang Yu commited on
Commit
158e9fd
1 Parent(s): a06fad0

Add kMaX-DeepLab

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ os.system("pip install gdown")
5
 
6
  os.system("pip install imutils")
7
 
8
- os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
9
 
10
  os.system("pip install git+https://github.com/cocodataset/panopticapi.git")
11
 
 
5
 
6
  os.system("pip install imutils")
7
 
8
+ os.system("python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'")
9
 
10
  os.system("pip install git+https://github.com/cocodataset/panopticapi.git")
11
 
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  pyyaml==5.1
2
- torch==1.9.0
3
- torchvision==0.10.0
4
 
5
  docutils==0.16
6
  # https://github.com/sphinx-doc/sphinx/commit/7acd3ada3f38076af7b2b5c9f3b60bb9c2587a3d
 
1
  pyyaml==5.1
2
+ torch==1.12.1
3
+ torchvision==0.13.1
4
 
5
  docutils==0.16
6
  # https://github.com/sphinx-doc/sphinx/commit/7acd3ada3f38076af7b2b5c9f3b60bb9c2587a3d