lkeab commited on
Commit
aa1da84
1 Parent(s): 5395ccf

update requirement

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  pyyaml==5.1
2
- torch
3
- torchvision
4
- opencv-python==4.4.0.40
5
  scikit-image
6
  kornia==0.5.11
7
 
1
  pyyaml==5.1
2
+ torch==1.7.1
3
+ torchvision==0.8.2
4
+ opencv-python-headless
5
  scikit-image
6
  kornia==0.5.11
7