fourier-convolutionspytorchfourierinpainting-methodsdeep-neural-networksinpainting-algorithmdeep-learninginpaintingcomputer-visionimage-inpaintingcolab-notebookhigh-resolutioncolabgenerative-adversarial-networkscnngenerative-adversarial-networkganfourier-transform
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
1.5 KiB
37 lines
1.5 KiB
mkdir celeba-hq-dataset |
|
|
|
unzip data256x256.zip -d celeba-hq-dataset/ |
|
|
|
# Reindex |
|
for i in `echo {00001..30000}` |
|
do |
|
mv 'celeba-hq-dataset/data256x256/'$i'.jpg' 'celeba-hq-dataset/data256x256/'$[10#$i - 1]'.jpg' |
|
done |
|
|
|
|
|
# Split: split train -> train & val |
|
cat fetch_data/train_shuffled.flist | shuf > celeba-hq-dataset/temp_train_shuffled.flist |
|
cat celeba-hq-dataset/temp_train_shuffled.flist | head -n 2000 > celeba-hq-dataset/val_shuffled.flist |
|
cat celeba-hq-dataset/temp_train_shuffled.flist | tail -n +2001 > celeba-hq-dataset/train_shuffled.flist |
|
cat fetch_data/val_shuffled.flist > celeba-hq-dataset/visual_test_shuffled.flist |
|
|
|
mkdir celeba-hq-dataset/train_256/ |
|
mkdir celeba-hq-dataset/val_source_256/ |
|
mkdir celeba-hq-dataset/visual_test_source_256/ |
|
|
|
cat celeba-hq-dataset/train_shuffled.flist | xargs -I {} mv celeba-hq-dataset/data256x256/{} celeba-hq-dataset/train_256/ |
|
cat celeba-hq-dataset/val_shuffled.flist | xargs -I {} mv celeba-hq-dataset/data256x256/{} celeba-hq-dataset/val_source_256/ |
|
cat celeba-hq-dataset/visual_test_shuffled.flist | xargs -I {} mv celeba-hq-dataset/data256x256/{} celeba-hq-dataset/visual_test_source_256/ |
|
|
|
|
|
# create location config celeba.yaml |
|
PWD=$(pwd) |
|
DATASET=${PWD}/celeba-hq-dataset |
|
CELEBA=${PWD}/configs/training/location/celeba.yaml |
|
|
|
touch $CELEBA |
|
echo "# @package _group_" >> $CELEBA |
|
echo "data_root_dir: ${DATASET}/" >> $CELEBA |
|
echo "out_root_dir: ${PWD}/experiments/" >> $CELEBA |
|
echo "tb_dir: ${PWD}/tb_logs/" >> $CELEBA |
|
echo "pretrained_models: ${PWD}/" >> $CELEBA
|
|
|