From d5216d60671a5669e1bd0ce2d6cc3e721d194390 Mon Sep 17 00:00:00 2001
From: wangna11BD <79366697+wangna11BD@users.noreply.github.com>
Date: Wed, 14 Sep 2022 21:47:33 +0800
Subject: [PATCH] fix tipc dataset for esrgan and StyleGANv2 (#689)

---
 benchmark/benchmark.yaml |  4 ++--
 test_tipc/prepare.sh     | 16 ++++++++--------
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/benchmark/benchmark.yaml b/benchmark/benchmark.yaml
index 844b776..3b1b9f0 100644
--- 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
index 386785d..0dcfac1 100644
--- 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*
-- 
GitLab