diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index 345fb9adc963f4a1a841e79ad5e13bb001636f01..1af5e4c3549e95eefc8aadb493d7ed09b65e970d 100644 --- a/test_tipc/prepare.sh +++ b/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