diff --git a/environment.yml b/environment.yml
index 6bd87e85a0416c21b57de84d9d6f540b9ac853fd..1897745ab16f6e3e26c89565e22d424645988795 100644
--- a/environment.yml
+++ b/environment.yml
@@ -1,162 +1,23 @@
 name: stcae
-channels:
-  - menpo
-  - conda-forge
-  - defaults
 dependencies:
-  - _libgcc_mutex=0.1=main
-  - _tflow_select=2.1.0=gpu
-  - absl-py=0.8.1=py36_0
-  - astor=0.8.0=py36_0
-  - attrs=19.3.0=py_0
-  - backcall=0.1.0=py36_0
-  - blas=1.0=mkl
-  - bleach=3.1.4=py_0
-  - bzip2=1.0.8=h7b6447c_0
-  - c-ares=1.15.0=h7b6447c_1001
-  - ca-certificates=2020.6.24=0
-  - cairo=1.14.12=h8948797_3
-  - certifi=2020.6.20=py36_0
-  - cudatoolkit=9.0=h13b8566_0
-  - cudnn=7.3.1=cuda9.0_0
-  - cupti=9.0.176=0
-  - cycler=0.10.0=py36_0
-  - dbus=1.13.12=h746ee38_0
-  - decorator=4.4.2=py_0
-  - defusedxml=0.6.0=py_0
-  - entrypoints=0.3=py36_0
-  - expat=2.2.6=he6710b0_0
-  - ffmpeg=4.2=h167e202_0
-  - fontconfig=2.13.0=h9420a91_0
-  - freetype=2.9.1=h8a8886c_1
-  - fribidi=1.0.5=h7b6447c_0
-  - gast=0.3.2=py_0
-  - glib=2.63.1=h5a9c865_0
-  - gmp=6.1.2=h6c8ec71_1
-  - gnutls=3.6.5=hd3a4fd2_1002
-  - graphite2=1.3.13=h23475e2_0
-  - graphviz=2.40.1=h21bd128_2
-  - grpcio=1.14.1=py36h9ba97e2_0
-  - gst-plugins-base=1.14.0=hbbd80ab_1
-  - gstreamer=1.14.0=hb453b48_1
-  - h5py=2.9.0=py36h7918eee_0
-  - harfbuzz=1.8.8=hffaf4a1_0
-  - hdf5=1.10.4=hb1b8bf9_0
-  - icu=58.2=h9c2bf20_1
-  - imbalanced-learn=0.5.0=py_0
-  - importlib_metadata=1.5.0=py36_0
-  - intel-openmp=2019.4=243
-  - ipykernel=5.1.4=py36h39e3cac_0
-  - ipython=7.13.0=py36h5ca1d4c_0
-  - ipython_genutils=0.2.0=py36_0
-  - ipywidgets=7.5.1=py_0
-  - jedi=0.16.0=py36_1
-  - jinja2=2.11.1=py_0
-  - joblib=0.14.0=py_0
-  - jpeg=9b=h024ee3a_2
-  - jsonschema=3.2.0=py36_0
-  - jupyter=1.0.0=py36_7
-  - jupyter_client=6.1.2=py_0
-  - jupyter_console=6.1.0=py_0
-  - jupyter_core=4.6.3=py36_0
-  - keras=2.2.2=0
-  - keras-applications=1.0.4=py36_1
-  - keras-base=2.2.2=py36_0
-  - keras-gpu=2.2.2=0
-  - keras-preprocessing=1.0.2=py36_1
-  - kiwisolver=1.1.0=py36he6710b0_0
-  - lame=3.100=h14c3975_1001
-  - libedit=3.1.20181209=hc058e9b_0
-  - libffi=3.2.1=hd88cf55_4
-  - libgcc-ng=9.1.0=hdf63c60_0
-  - libgfortran-ng=7.3.0=hdf63c60_0
-  - libiconv=1.15=h63c8f33_5
-  - libpng=1.6.37=hbc83047_0
-  - libprotobuf=3.10.1=hd408876_0
-  - libsodium=1.0.16=h1bed415_0
-  - libstdcxx-ng=9.1.0=hdf63c60_0
-  - libtiff=4.1.0=h2733197_0
-  - libuuid=1.0.3=h1bed415_2
-  - libxcb=1.13=h1bed415_1
-  - libxml2=2.9.9=hea5a465_1
-  - llvmlite=0.30.0=py36hd408876_0
-  - markdown=3.1.1=py36_0
-  - markupsafe=1.1.1=py36h7b6447c_0
-  - matplotlib=3.1.1=py36h5429711_0
-  - mistune=0.8.4=py36h7b6447c_0
-  - mkl=2019.4=243
-  - mkl-service=2.3.0=py36he904b0f_0
-  - mkl_fft=1.0.15=py36ha843d7b_0
-  - mkl_random=1.1.0=py36hd6b4f25_0
-  - natsort=7.0.1=py_0
-  - nbconvert=5.6.1=py36_0
-  - nbformat=5.0.4=py_0
-  - ncurses=6.1=he6710b0_1
-  - nettle=3.4.1=h1bed415_1002
-  - notebook=6.0.3=py36_0
-  - numba=0.46.0=py36h962f231_0
-  - numpy=1.17.4=py36hc1035e2_0
-  - numpy-base=1.17.4=py36hde5b4d6_0
-  - opencv3=3.1.0=py36_0
-  - openh264=1.8.0=hdbcaa40_1000
-  - openssl=1.0.2u=h7b6447c_0
-  - pandas=0.25.3=py36he6710b0_0
-  - pandoc=2.2.3.2=0
-  - pandocfilters=1.4.2=py36_1
-  - pango=1.42.4=h049681c_0
-  - parso=0.6.2=py_0
-  - pcre=8.43=he6710b0_0
-  - pexpect=4.8.0=py36_0
-  - pickleshare=0.7.5=py36_0
-  - pip=19.3.1=py36_0
-  - pixman=0.38.0=h7b6447c_0
-  - prometheus_client=0.7.1=py_0
-  - prompt-toolkit=3.0.4=py_0
-  - prompt_toolkit=3.0.4=0
-  - protobuf=3.10.1=py36he6710b0_0
-  - ptyprocess=0.6.0=py36_0
-  - pydot=1.4.1=py36_0
-  - pygments=2.6.1=py_0
-  - pyparsing=2.4.5=py_0
-  - pyqt=5.9.2=py36h05f1152_2
-  - pyrsistent=0.16.0=py36h7b6447c_0
-  - python=3.6.4=hc3d631a_3
-  - python-dateutil=2.8.1=py_0
-  - pytz=2019.3=py_0
-  - pyyaml=5.1.2=py36h7b6447c_0
-  - pyzmq=18.1.1=py36he6710b0_0
-  - qt=5.9.6=h8703b6f_2
-  - qtconsole=4.7.3=py_0
-  - qtpy=1.9.0=py_0
-  - readline=7.0=h7b6447c_5
-  - scikit-learn=0.21.3=py36hd81dba3_0
-  - scipy=1.3.1=py36h7c811a0_0
-  - send2trash=1.5.0=py36_0
-  - setuptools=42.0.1=py36_0
-  - sip=4.19.8=py36hf484d3e_0
-  - six=1.13.0=py36_0
-  - sqlite=3.30.1=h7b6447c_0
-  - tbb=2019.8=hfd86e86_0
-  - tensorboard=1.10.0=py36hf484d3e_0
-  - tensorflow=1.10.0=gpu_py36h97a2126_0
-  - tensorflow-base=1.10.0=gpu_py36h6ecc378_0
-  - tensorflow-gpu=1.10.0=hf154084_0
-  - termcolor=1.1.0=py36_1
-  - terminado=0.8.3=py36_0
-  - testpath=0.4.4=py_0
-  - tk=8.6.8=hbc83047_0
-  - tornado=6.0.3=py36h7b6447c_0
-  - traitlets=4.3.3=py36_0
-  - wcwidth=0.1.9=py_0
-  - webencodings=0.5.1=py36_1
-  - werkzeug=0.16.0=py_0
-  - wheel=0.33.6=py36_0
-  - widgetsnbextension=3.5.1=py36_0
-  - x264=1!152.20180806=h14c3975_0
-  - xz=5.2.4=h14c3975_4
-  - yaml=0.1.7=had09818_2
-  - zeromq=4.3.1=he6710b0_3
-  - zipp=2.2.0=py_0
-  - zlib=1.2.11=h7b6447c_3
-  - zstd=1.3.7=h0b5b093_0
-
+  - python=3.6.4
+  - tensorflow=1.10.0
+  - tensorflow-gpu=1.10.0
+  - hdf5
+  - h5py
+  - graphviz
+  - pydot
+  - keras=2.2.2
+  - keras-gpu=2.2.2
+  - matplotlib
+  - pandas
+  - scikit-learn
+  - ffmpeg
+  - numba
+  - cudatoolkit=9.0
+  - cudnn=7.3.1
+  - jupyter
+  - natsort
+  - pip:
+    - opencv-python>=3.3.1
+    - imbalanced-learn