fix whole indfer (#600)

test_tipc/prepare.sh

@@ -74,11 +74,11 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
         mv ./data/SinGAN-official_images/Images/stone.png ./data/singan ;;
     esac
 elif [ ${MODE} = "whole_train_whole_infer" ];then
-    if [ ${model_name} == "pix2pix" ]; then
+    if [ ${model_name} == "Pix2pix" ]; then
         rm -rf ./data/facades*
         wget -nc -P ./data/ http://efrosgans.eecs.berkeley.edu/pix2pix/datasets/facades.tar.gz --no-check-certificate
         cd ./data/ && tar -xzf facades.tar.gz && cd ../
-    elif [ ${model_name} == "cyclegan" ]; then
+    elif [ ${model_name} == "CycleGAN" ]; then
         rm -rf ./data/horse2zebra*
         wget -nc -P ./data/ https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/horse2zebra.zip --no-check-certificate
         cd ./data/ && unzip horse2zebra.zip && cd ../
@@ -90,19 +90,19 @@ elif [ ${MODE} = "whole_train_whole_infer" ];then
         mv ./data/SinGAN-official_images/Images/stone.png ./data/singan
     fi
 elif [ ${MODE} = "lite_train_whole_infer" ];then
-    if [ ${model_name} == "pix2pix" ]; then
+    if [ ${model_name} == "Pix2pix" ]; then
         rm -rf ./data/facades*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/pix2pix_facade_lite.tar --no-check-certificate
         cd ./data/ && tar xf pix2pix_facade_lite.tar && cd ../
-    elif [ ${model_name} == "cyclegan" ]; then
+    elif [ ${model_name} == "CycleGAN" ]; then
         rm -rf ./data/horse2zebra*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/cyclegan_horse2zebra_lite.tar  --no-check-certificate --no-check-certificate
         cd ./data/ && tar xf cyclegan_horse2zebra_lite.tar && cd ../
-    elif [ ${model_name} == "fom" ]; then
+    elif [ ${model_name} == "FOMM" ]; then
         rm -rf ./data/first_order*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/fom_lite.tar  --no-check-certificate --no-check-certificate
         cd ./data/ && tar xf fom_lite.tar && cd ../
-    elif [ ${model_name} == "stylegan2" ]; then
+    elif [ ${model_name} == "StyleGANv2" ]; then
         rm -rf ./data/ffhq*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq.tar --no-check-certificate
         cd ./data/ && tar xf ffhq.tar && cd ../
@@ -122,28 +122,28 @@ elif [ ${MODE} = "lite_train_whole_infer" ];then
         mv ./data/SinGAN-official_images/Images/stone.png ./data/singan
     fi
 elif [ ${MODE} = "whole_infer" ];then
-    if [ ${model_name} = "pix2pix" ]; then
+    if [ ${model_name} = "Pix2pix" ]; then
         rm -rf ./data/facades*
         rm -rf ./inference/pix2pix*
         wget -nc  -P ./inference https://paddlegan.bj.bcebos.com/static_model/pix2pix_facade.tar --no-check-certificate
         wget -nc  -P ./data https://paddlegan.bj.bcebos.com/datasets/facades_test.tar --no-check-certificate
         cd ./data && tar xf facades_test.tar && mv facades_test facades && cd ../
         cd ./inference && tar xf pix2pix_facade.tar && cd ../
-    elif [ ${model_name} = "cyclegan" ]; then
+    elif [ ${model_name} = "CycleGAN" ]; then
         rm -rf ./data/cyclegan*
         rm -rf ./inference/cyclegan*
         wget -nc  -P ./inference https://paddlegan.bj.bcebos.com/static_model/cyclegan_horse2zebra.tar  --no-check-certificate
         wget -nc  -P ./data https://paddlegan.bj.bcebos.com/datasets/cyclegan_horse2zebra_test.tar  --no-check-certificate
         cd ./data && tar xf cyclegan_horse2zebra_test.tar && mv cyclegan_test horse2zebra && cd ../
         cd ./inference && tar xf cyclegan_horse2zebra.tar && cd ../
-    elif [ ${model_name} == "fom" ]; then
+    elif [ ${model_name} == "FOMM" ]; then
         rm -rf ./data/first_order*
         rm -rf ./inference/fom_dy2st*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/fom_lite_test.tar  --no-check-certificate
         wget -nc  -P ./inference https://paddlegan.bj.bcebos.com/static_model/fom_dy2st.tar --no-check-certificate
         cd ./data/ && tar xf fom_lite_test.tar && cd ../
         cd ./inference && tar xf fom_dy2st.tar && cd ../
-    elif [ ${model_name} == "stylegan2" ]; then
+    elif [ ${model_name} == "StyleGANv2" ]; then
         rm -rf ./data/ffhq*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq.tar --no-check-certificate
         wget -nc  -P ./inference https://paddlegan.bj.bcebos.com/static_model/stylegan2_1024.tar --no-check-certificate