diff --git a/research/cv/wgan/infer/Dockerfile b/research/cv/wgan/Dockerfile
similarity index 100%
rename from research/cv/wgan/infer/Dockerfile
rename to research/cv/wgan/Dockerfile
diff --git a/research/cv/wgan/docker_start.sh b/research/cv/wgan/docker_start.sh
new file mode 100644
index 0000000000000000000000000000000000000000..9fdd0e872e3d2d5b8a3408623f9c53ad7e86e03c
--- /dev/null
+++ b/research/cv/wgan/docker_start.sh
@@ -0,0 +1,29 @@
+#!/bin/bash
+
+docker_image=$1
+data_dir=$2
+model_dir=$3
+
+docker run -it --ipc=host \
+               --device=/dev/davinci0 \
+               --device=/dev/davinci1 \
+               --device=/dev/davinci2 \
+               --device=/dev/davinci3 \
+               --device=/dev/davinci4 \
+               --device=/dev/davinci5 \
+               --device=/dev/davinci6 \
+               --device=/dev/davinci7 \
+               --device=/dev/davinci_manager \
+               --device=/dev/devmm_svm \
+               --device=/dev/hisi_hdc \
+               --privileged \
+               -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
+               -v /usr/local/Ascend/add-ons/:/usr/local/Ascend/add-ons \
+               -v ${data_dir}:${data_dir} \
+               -v ${model_dir}:${model_dir} \
+               -v /var/log/npu/conf/slog/slog.conf:/var/log/npu/conf/slog/slog.conf \
+               -v /var/log/npu/slog/:/var/log/npu/slog/ \
+               -v /var/log/npu/profiling/:/var/log/npu/profiling \
+               -v /var/log/npu/dump/:/var/log/npu/dump \
+               -v /var/log/npu/:/usr/slog ${docker_image} \
+               /bin/bash
diff --git a/research/cv/wgan/infer/requirements.txt b/research/cv/wgan/infer/requirements.txt
deleted file mode 100644
index 2006ec1e3378ba2f3017908d93176613c45ed839..0000000000000000000000000000000000000000
--- a/research/cv/wgan/infer/requirements.txt
+++ /dev/null
@@ -1,2 +0,0 @@
-Pillow
-lmdb
diff --git a/research/cv/wgan/requirements.txt b/research/cv/wgan/requirements.txt
index 5ae5f9e3a954e7a7d36001d90aed700253b0949d..2006ec1e3378ba2f3017908d93176613c45ed839 100644
--- a/research/cv/wgan/requirements.txt
+++ b/research/cv/wgan/requirements.txt
@@ -1,3 +1,2 @@
-mindspore
-numpy
-Pillow
\ No newline at end of file
+Pillow
+lmdb