未验证 提交 d5216d60 编写于 作者: W wangna11BD 提交者: GitHub

fix tipc dataset for esrgan and StyleGANv2 (#689)

上级 c0039d13
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
......
......@@ -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*
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册