Fix pip requirements issues

5 years ago · faa6aeaa20
parent 099e6f5ebd
commit faa6aeaa20
1 changed files with 3 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,4 +1,5 @@
 # pip install -U -r requirements.txt
+Cython
 numpy==1.17
 opencv-python
 torch >= 1.5
@ -8,7 +9,8 @@ tqdm
 pillow
 tensorboard
 pyyaml
-
+torchvision
+scipy

 # Nvidia Apex (optional) for mixed precision training --------------------------
 # git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex