Removing scipy as dependency

Author: pushkalkatara

Dockerfile

@@ -21,7 +21,7 @@ ENV LIBRARY_PATH /usr/local/nvidia/lib64:/usr/local/nvidia/lib:/usr/local/cuda/l
 
 # python3 modules
 RUN wget https://bootstrap.pypa.io/get-pip.py && python3 get-pip.py && \
-        pip3 install --upgrade --no-cache-dir wheel six setuptools cython numpy scipy==1.2.0 \
+        pip3 install --upgrade --no-cache-dir wheel six setuptools cython numpy imageio \
                 matplotlib seaborn scikit-learn scikit-image pillow requests \
                 jupyterlab networkx h5py pandas plotly protobuf tqdm tensorboardX colorama setproctitle && \
         pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp35-cp35m-linux_x86_64.whl

README.md

@@ -48,7 +48,7 @@ L1 and L2 losses with multi-scale support are available in [losses.py](./losses.
 Currently, the code supports python 3
 * numpy 
 * PyTorch ( == 0.4.1, for <= 0.4.0 see branch [python36-PyTorch0.4](https://github.com/NVIDIA/flownet2-pytorch/tree/python36-PyTorch0.4))
-* scipy 
+* imageio
 * scikit-image
 * tensorboardX
 * colorama, tqdm, setproctitle 

datasets.py

@@ -8,7 +8,7 @@
 from glob import glob
 import utils.frame_utils as frame_utils
 
-from scipy.misc import imread, imresize
+from imageio import imread
 
 class StaticRandomCrop(object):
     def __init__(self, image_size, crop_size):

utils/frame_utils.py

@@ -1,6 +1,6 @@
 import numpy as np
 from os.path import *
-from scipy.misc import imread
+from imageio import imread
 from . import flow_utils 
 
 def read_gen(file_name):