Miroslav Purkrabek commited on
Commit
f70b212
·
1 Parent(s): 321ec31

revert to CPU-based NMS

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -5
requirements.txt CHANGED
@@ -1,11 +1,9 @@
1
- torch==2.4.0
2
- torchvision>=0.16.2
3
  numpy==1.25.1
4
  opencv-python==4.9.0.80
5
 
6
- # mmcv wheel matching Torch 2.4 + cu124
7
- -f https://mmwheels.onedl.ai/cu124-torch240/index.html
8
- mmcv==2.3.0-rc0
9
  mmdet==3.3.0
10
  mmpretrain==1.2.0
11
 
 
1
+ torch==2.1.2
2
+ torchvision==0.16.2
3
  numpy==1.25.1
4
  opencv-python==4.9.0.80
5
 
6
+ mmcv==2.1.0
 
 
7
  mmdet==3.3.0
8
  mmpretrain==1.2.0
9