From e53c593f25440f62726319f877aa4af930ef8e3e Mon Sep 17 00:00:00 2001 From: Wang Xin Date: Fri, 24 Nov 2023 18:47:23 +0800 Subject: [PATCH] add ConSinGAN model (#2) --- CONSINGAN/start_export.sh | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 CONSINGAN/start_export.sh diff --git a/CONSINGAN/start_export.sh b/CONSINGAN/start_export.sh new file mode 100644 index 0000000..8dcf2d0 --- /dev/null +++ b/CONSINGAN/start_export.sh @@ -0,0 +1,17 @@ +ROOT=`pwd` + +pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 +pip install scipy scikit-image matplotlib albumentations + + +git clone https://github.com/PaddleJitLab/ConSinGAN +cd ConSinGAN +git checkout master +python export.py --train_mode generation --input_name Images/Generation/angkorwat.jpg + + +cd $ROOT +git clone https://github.com/PaddleJitLab/ConSinGAN.paddle +cd ConSinGAN.paddle +git checkout main +python export.py --train_mode generation --input_name Images/Generation/angkorwat.jpg