wellton/stargan
1
0
Fork 0
Código Issues Pull Requests Ações Pacotes Projetos Versões Wiki Atividade

fixed the download link

Ver Código-Fonte
Esse commit está contido em:
yunjey
2019-01-22 11:16:52 +00:00
pai dacdddbbfc
commit f4a1c5873f
2 arquivos alterados com 5 adições e 11 exclusões
Baixar Arquivo de Patch Baixar Arquivo de Diff Expandir todos os arquivos Colapsar todos os arquivos
download.sh
+3 -9
Ver Arquivo
@@ -2,20 +2,14 @@ FILE=$1
if [ $FILE == "celeba" ]; then if [ $FILE == "celeba" ]; then
# CelebA images # CelebA images and attribute labels
URL=https://www.dropbox.com/s/3e5cmqgplchz85o/CelebA_nocrop.zip?dl=0 URL=https://www.dropbox.com/s/d1kjpkqklf0uw77/celeba.zip?dl=0
ZIP_FILE=./data/CelebA_nocrop.zip ZIP_FILE=./data/celeba.zip
mkdir -p ./data/ mkdir -p ./data/
wget -N $URL -O $ZIP_FILE wget -N $URL -O $ZIP_FILE
unzip $ZIP_FILE -d ./data/ unzip $ZIP_FILE -d ./data/
rm $ZIP_FILE rm $ZIP_FILE
# CelebA attribute labels
URL=https://www.dropbox.com/s/auexdy98c6g7y25/list_attr_celeba.zip?dl=0
ZIP_FILE=./data/list_attr_celeba.zip
wget -N $URL -O $ZIP_FILE
unzip $ZIP_FILE -d ./data/
rm $ZIP_FILE
elif [ $FILE == 'pretrained-celeba-128x128' ]; then elif [ $FILE == 'pretrained-celeba-128x128' ]; then
main.py
+2 -2
Ver Arquivo
@@ -91,8 +91,8 @@ if __name__ == '__main__':
parser.add_argument('--use_tensorboard', type=str2bool, default=True) parser.add_argument('--use_tensorboard', type=str2bool, default=True)
# Directories. # Directories.
parser.add_argument('--celeba_image_dir', type=str, default='data/CelebA_nocrop/images') parser.add_argument('--celeba_image_dir', type=str, default='data/celeba/images')
parser.add_argument('--attr_path', type=str, default='data/list_attr_celeba.txt') parser.add_argument('--attr_path', type=str, default='data/celeba/list_attr_celeba.txt')
parser.add_argument('--rafd_image_dir', type=str, default='data/RaFD/train') parser.add_argument('--rafd_image_dir', type=str, default='data/RaFD/train')
parser.add_argument('--log_dir', type=str, default='stargan/logs') parser.add_argument('--log_dir', type=str, default='stargan/logs')
parser.add_argument('--model_save_dir', type=str, default='stargan/models') parser.add_argument('--model_save_dir', type=str, default='stargan/models')