Skip to content

P
PaddleGAN

PaddlePaddle / PaddleGAN
1 年多 前同步成功

通知 97
Star 7254
Fork 1210
P
PaddleGAN
未验证 提交 d5216d60 编写于 作者: W wangna11BD 提交者： GitHub
浏览文件
操作

fix tipc dataset for esrgan and StyleGANv2 (#689)

上级 c0039d13
显示空白变更内容
内联 并排
Showing with 10 addition and 10 deletion
benchmark/benchmark.yaml
浏览文件 @ d5216d60
StyleGANv2: 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 config: configs/stylegan_v2_256_ffhq.yaml
fp_item: fp32 fp_item: fp32
bs_item: 8 bs_item: 8
...@@ -15,7 +15,7 @@ FOMM: ...@@ -15,7 +15,7 @@ FOMM:
log_interval: 1 log_interval: 1
esrgan: 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 config: configs/esrgan_psnr_x4_div2k.yaml
fp_item: fp32 fp_item: fp32
bs_item: 32 bs_item: 32
......
test_tipc/prepare.sh
浏览文件 @ d5216d60
...@@ -48,8 +48,8 @@ if [ ${MODE} = "lite_train_lite_infer" ];then ...@@ -48,8 +48,8 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
cd ./data/ && tar xf cyclegan_horse2zebra_lite.tar && cd ../ ;; cd ./data/ && tar xf cyclegan_horse2zebra_lite.tar && cd ../ ;;
StyleGANv2) StyleGANv2)
rm -rf ./data/ffhq* 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
cd ./data/ && tar xf ffhq.tar && cd ../ ;; cd ./data/ && tar xf ffhq_256.tar && cd ../ ;;
FOMM) FOMM)
rm -rf ./data/fom_lite* rm -rf ./data/fom_lite*
wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/fom_lite.tar --no-check-certificate --no-check-certificate 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 ...@@ -60,8 +60,8 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
cd ./data/ && tar xf reds_lite.tar && cd ../ ;; cd ./data/ && tar xf reds_lite.tar && cd ../ ;;
esrgan) esrgan)
rm -rf ./data/DIV2K* rm -rf ./data/DIV2K*
wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14.tar --no-check-certificate wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14paddle.tar --no-check-certificate
cd ./data/ && tar xf DIV2KandSet14.tar && cd ../ ;; cd ./data/ && tar xf DIV2KandSet14paddle.tar && cd ../ ;;
singan) singan)
rm -rf ./data/singan* rm -rf ./data/singan*
wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/singan-official_images.zip --no-check-certificate 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 ...@@ -100,8 +100,8 @@ elif [ ${MODE} = "lite_train_whole_infer" ];then
cd ./data/ && tar xf fom_lite.tar && cd ../ cd ./data/ && tar xf fom_lite.tar && cd ../
elif [ ${model_name} == "StyleGANv2" ]; then elif [ ${model_name} == "StyleGANv2" ]; then
rm -rf ./data/ffhq* 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
cd ./data/ && tar xf ffhq.tar && cd ../ cd ./data/ && tar xf ffhq_256.tar && cd ../
elif [ ${model_name} == "basicvsr" ]; then elif [ ${model_name} == "basicvsr" ]; then
rm -rf ./data/reds* rm -rf ./data/reds*
wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/reds_lite.tar --no-check-certificate 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 ...@@ -141,10 +141,10 @@ elif [ ${MODE} = "whole_infer" ];then
cd ./inference && tar xf fom_dy2st.tar && cd ../ cd ./inference && tar xf fom_dy2st.tar && cd ../
elif [ ${model_name} == "StyleGANv2" ]; then elif [ ${model_name} == "StyleGANv2" ]; then
rm -rf ./data/ffhq* 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 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 ./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 elif [ ${model_name} == "basicvsr" ]; then
rm -rf ./data/reds* rm -rf ./data/reds*
rm -rf ./inference/basic* rm -rf ./inference/basic*
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或