diff --git a/ai-box/Dockerfile b/ai-box/Dockerfile
index 55b77b1e36def970700bf497978e96809643bbb2..456662d35d26b4c2896810f938146e114b822a56 100644
--- a/ai-box/Dockerfile
+++ b/ai-box/Dockerfile
@@ -14,7 +14,7 @@
 # limitations under the License.
 #
 
-FROM registry-vpc.cn-hangzhou.aliyuncs.com/schbrain/python:3.7.0
+FROM registry-vpc.cn-hangzhou.aliyuncs.com/schbrain/python:3.6.7
 MAINTAINER xt
 
 ENV PYTHONUNBUFFERED 1
@@ -25,7 +25,6 @@ ENV LIBRARY_PATH /usr/lib/gcc/x86_64-linux-gnu/8
 COPY sources.list /etc/apt/sources.list
 COPY requirements.txt ./
 COPY simsun.ttc /usr/local/fonts/
-COPY tensorflow-1.12.0-cp37-cp37m-linux_x86_64.whl ./
 RUN apt-key adv --recv-keys --keyserver keyserver.ubuntu.com 40976EAF437D05B5 \
     && apt-key adv --recv-keys --keyserver keyserver.ubuntu.com 3B4FE6ACC0B21F32 \
     && apt-get update \
@@ -35,7 +34,6 @@ RUN apt-key adv --recv-keys --keyserver keyserver.ubuntu.com 40976EAF437D05B5 \
          r-base=3.5.1-* r-base-dev=3.5.1-* r-recommended=3.5.1-* \
     && pip install --no-cache-dir --trusted-host mirrors.aliyun.com \
          -i http://mirrors.aliyun.com/pypi/simple -r requirements.txt \
-    && pip install ./tensorflow-1.12.0-cp37-cp37m-linux_x86_64.whl \
     && echo 'options(repos = c(CRAN = "https://cloud.r-project.org/"))' >> /etc/R/Rprofile.site \
     && echo 'source("/etc/R/Rprofile.site")' >> /etc/littler.r \
     && ln -s /usr/lib/R/site-library/littler/examples/install.r /usr/local/bin/install.r \
@@ -45,7 +43,6 @@ RUN apt-key adv --recv-keys --keyserver keyserver.ubuntu.com 40976EAF437D05B5 \
     && install.r docopt \
     && install.r mirt \
     && install.r mirtCAT \
-    && rm ./tensorflow-1.12.0-cp37-cp37m-linux_x86_64.whl \
     && rm -rf /tmp/downloaded_packages/ /tmp/*.rds \
     && rm -rf /var/lib/apt/lists/*
 
diff --git a/ai-box/requirements.txt b/ai-box/requirements.txt
index 11d076e6a5b2015b21db563f352328012ffdc6b8..ff8476af21e38dd97a1c1ed488e293dd0ce04de4 100644
--- a/ai-box/requirements.txt
+++ b/ai-box/requirements.txt
@@ -27,6 +27,7 @@ vine==1.1.4
 xlrd==1.1.0
 scipy
 scikit-learn
+scikit-image==0.13.1
 matplotlib
 pandas
 numpy
@@ -39,4 +40,5 @@ oss2==2.5.0
 zbarlight==2.1
 imutils==0.5.1
 seaborn==0.8.1
+tensorflow==1.12.0
 keras==2.2.4