Removed dependency on Kornia library

jac99 · Mar 6, 2022 · e9bb28d · e9bb28d
1 parent 2b50b72
commit e9bb28d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -41,7 +41,8 @@ If you find our work useful, please consider citing:
 
 ### Environment and Dependencies
 Code was tested using Python 3.8 with PyTorch 1.10.2 on Ubuntu 18.04 with CUDA 10.2.
-Other dependencies include:
+Dependencies:
+* PyTorch (version 1.10 or above) 
 * opencv-python
 * scipy
 * PIL