fix tipc dataset for esrgan and StyleGANv2 (#689)

PaddlePaddle · Sep 14, 2022 · d5216d6 · d5216d6
1 parent c0039d1
commit d5216d6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/benchmark/benchmark.yaml b/benchmark/benchmark.yaml
@@ -1,5 +1,5 @@
 StyleGANv2:
-  dataset_web: https://paddlegan.bj.bcebos.com/datasets/ffhq.tar
+  dataset_web: https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar
   config: configs/stylegan_v2_256_ffhq.yaml
   fp_item: fp32
   bs_item: 8
@@ -15,7 +15,7 @@ FOMM:
   log_interval: 1
 
 esrgan:
-  dataset_web: https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14.tar
+  dataset_web: https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14paddle.tar
   config: configs/esrgan_psnr_x4_div2k.yaml
   fp_item: fp32
   bs_item: 32

diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
@@ -48,8 +48,8 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
         cd ./data/ && tar xf cyclegan_horse2zebra_lite.tar && cd ../ ;;
     StyleGANv2)
         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 ../ ;;
+        wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar --no-check-certificate
+        cd ./data/ && tar xf ffhq_256.tar && cd ../ ;;
     FOMM)
         rm -rf ./data/fom_lite*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/fom_lite.tar  --no-check-certificate --no-check-certificate
@@ -60,8 +60,8 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
         cd ./data/ && tar xf reds_lite.tar && cd ../ ;;
     esrgan)
         rm -rf ./data/DIV2K*
-        wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14.tar --no-check-certificate
-        cd ./data/ && tar xf DIV2KandSet14.tar && cd ../ ;;
+        wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14paddle.tar --no-check-certificate
+        cd ./data/ && tar xf DIV2KandSet14paddle.tar && cd ../ ;;
     singan)
         rm -rf ./data/singan*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/singan-official_images.zip --no-check-certificate
@@ -100,8 +100,8 @@ elif [ ${MODE} = "lite_train_whole_infer" ];then
         cd ./data/ && tar xf fom_lite.tar && cd ../
     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 ../
+        wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar --no-check-certificate
+        cd ./data/ && tar xf ffhq_256.tar && cd ../
     elif [ ${model_name} == "basicvsr" ]; then
         rm -rf ./data/reds*
         wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/reds_lite.tar --no-check-certificate
@@ -141,10 +141,10 @@ elif [ ${MODE} = "whole_infer" ];then
         cd ./inference && tar xf fom_dy2st.tar && cd ../
     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 ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar --no-check-certificate
         wget -nc  -P ./inference https://paddlegan.bj.bcebos.com/static_model/stylegan2_1024.tar --no-check-certificate
         cd ./inference && tar xf stylegan2_1024.tar && cd ../
-        cd ./data/ && tar xf ffhq.tar && cd ../
+        cd ./data/ && tar xf ffhq_256.tar && cd ../
     elif [ ${model_name} == "basicvsr" ]; then
         rm -rf ./data/reds*
         rm -rf ./inference/basic*