diff --git a/model_zoo/research/cv/wgan/export.py b/model_zoo/research/cv/wgan/export.py
index 0234c1ffa76..1d9e0d00764 100644
--- a/model_zoo/research/cv/wgan/export.py
+++ b/model_zoo/research/cv/wgan/export.py
@@ -23,7 +23,8 @@ import mindspore.common.dtype as mstype
 from mindspore import Tensor, load_checkpoint, load_param_into_net, export, context
 
 from src.args import get_args
-
+from src.dcgan_model import DcganG
+from src.dcgannobn_model import DcgannobnG
 
 if __name__ == '__main__':
     args_opt = get_args('export')
diff --git a/model_zoo/research/cv/wgan/scripts/run_infer_310.sh b/model_zoo/research/cv/wgan/scripts/run_infer_310.sh
index 8e824846dcc..60d327516b7 100644
--- a/model_zoo/research/cv/wgan/scripts/run_infer_310.sh
+++ b/model_zoo/research/cv/wgan/scripts/run_infer_310.sh
@@ -102,6 +102,10 @@ function infer()
 
 function cal_acc()
 {
+    if [ -d infer_output ]; then
+        rm -rf ./infer_output
+    fi
+    mkdir infer_output
     python3.7 ../postprocess.py --config=$config_path --output_dir=./infer_output --nimages=$nimages --post_result_path=./result_Files --device_id=$device_id &> acc.log
 }