wgan export bug fix master

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')

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
 }