mirror of https://github.com/menyifang/DCT-Net
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.3 KiB
Bash
45 lines
1.3 KiB
Bash
|
|
data_root='data'
|
|
align_dir='raw_style_data_faces'
|
|
|
|
echo "STEP: start to prepare data for stylegan ..."
|
|
cd $data_root
|
|
if [ ! -d stylegan ]; then
|
|
mkdir stylegan
|
|
fi
|
|
cd stylegan
|
|
stylegan_data_dir=$(pwd)
|
|
if [ ! -d "$(date +"%Y%m%d")" ]; then
|
|
mkdir "$(date +"%Y%m%d")"
|
|
fi
|
|
cd "$(date +"%Y%m%d")"
|
|
cp $align_dir . -r
|
|
if [ -d $(echo $align_dir) ]; then
|
|
cp $(echo $align_dir) . -r
|
|
fi
|
|
src_dir_sg=$(pwd)
|
|
|
|
cd $data_root/../source
|
|
outdir_sg="$(echo $stylegan_data_dir)/traindata_$(echo $stylename)_256_$(date +"%m%d")"
|
|
echo $outdir_sg
|
|
echo $src_dir_sg
|
|
if [ ! -d "$outdir_sg" ]; then
|
|
python prepare_data.py --size 256 --out $outdir_sg $src_dir_sg
|
|
fi
|
|
echo "prepare data for stylegan finished!"
|
|
|
|
### train model
|
|
#cd $data_root
|
|
#cd stylegan
|
|
#stylegan_data_dir=$(pwd)
|
|
#outdir_sg="$(echo $stylegan_data_dir)/traindata_$(echo $stylename)_256_$(date +"%m%d")"
|
|
#echo "STEP:start to train the style learner ..."
|
|
#echo $outdir_sg
|
|
#exp_name="ffhq_$(echo $stylename)_s256_id01_$(date +"%m%d")"
|
|
#cd /data/vdb/qingyao/cartoon/mycode/stylegan2-pytorch
|
|
#model_path=face_generation/experiment_stylegan/$(echo $exp_name)/models/001000.pt
|
|
#if [ ! -f "$model_path" ]; then
|
|
# CUDA_VISIBLE_DEVICES=6 python train_condition.py --name $exp_name --path $outdir_sg --config config/conf_server_train_condition_shell.json
|
|
#fi
|
|
#### [training...]
|
|
#echo "train the style learner finished!" |