Skip to content

Commit d5da778

Browse files
committed
simplified celebagen script
1 parent 058d084 commit d5da778

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

fetch_data/celebahq_dataset_prepare.sh

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -9,22 +9,19 @@ do
99
done
1010

1111

12-
# train/test/vis split
12+
# Split: split train -> train & val
1313
cat fetch_data/train_shuffled.flist | shuf > celeba-hq-dataset/temp_train_shuffled.flist
1414
cat celeba-hq-dataset/temp_train_shuffled.flist | head -n 2000 > celeba-hq-dataset/val_shuffled.flist
1515
cat celeba-hq-dataset/temp_train_shuffled.flist | tail -n +2001 > celeba-hq-dataset/train_shuffled.flist
16-
1716
cat fetch_data/val_shuffled.flist > celeba-hq-dataset/visual_test_shuffled.flist
1817

18+
mkdir celeba-hq-dataset/train_256/
19+
mkdir celeba-hq-dataset/val_source_256/
20+
mkdir celeba-hq-dataset/visual_test_source_256/
1921

20-
# Split
21-
for mode in train \
22-
val \
23-
visual_test
24-
do
25-
mkdir celeba-hq-dataset/$mode"_source_256/"
26-
cat celeba-hq-dataset/$mode"_shuffled.flist" | xargs -I {} mv celeba-hq-dataset/data256x256/{} celeba-hq-dataset/$mode"_source_256/"
27-
done
22+
cat celeba-hq-dataset/train_shuffled.flist | xargs -I {} mv celeba-hq-dataset/data256x256/{} celeba-hq-dataset/train_256/
23+
cat celeba-hq-dataset/val_shuffled.flist | xargs -I {} mv celeba-hq-dataset/data256x256/{} celeba-hq-dataset/val_source_256/
24+
cat celeba-hq-dataset/visual_test_shuffled.flist | xargs -I {} mv celeba-hq-dataset/data256x256/{} celeba-hq-dataset/visual_test_source_256/
2825

2926

3027
# create location config celeba.yaml

0 commit comments

Comments
 (0)