Author SHA1 Message Date
luni 25c108fe83 get error 2023-01-27 14:08:39 +00:00
luni 970aeaf5b9 calculate error 2023-01-26 19:23:37 +00:00
luni bda4582c3c run evaluation 2023-01-26 16:20:39 +00:00
luni 0bc5633b6c retraining original abl_all_post 2023-01-25 22:30:46 +00:00
luni ffdc60062d add helpers 2023-01-25 14:37:36 +00:00
luni 73d374401b outdated 2023-01-25 13:07:24 +00:00
7 changed files with 226 additions and 59 deletions
-49
View File
@@ -1,49 +0,0 @@
# stylegan2-tf-2.x
* This is tensorflow 2.0 based keras subclassing reimplementation of official [StyleGAN2 Repo](https://github.com/NVlabs/stylegan2)
## Inference from official weights
* Please check [how_to_extract_official_weights.md](./how_to_extract_official_weights.md)
| Official repo result | Official weight copied result from this repo |
| :---: | :---: |
| ![official_result] | ![restored_result] |
## Training on 256x256
* batch size 32
* around 300k steps
| d_loss | g_loss |
| :---: | :---: |
| ![256x256_d_loss]| ![256x256_g_loss]|
* result samples
| | | |
| :---: | :---: | :---: |
| ![256x256_result_0]| ![256x256_result_1]| ![256x256_result_2]|
## Using custom data
* Create dataset(tf.data.Dataset) loader which returns
* shape: `[None, 3, res, res]` where `res` is desired image resolution
* image value range: `[-1, +1]`
## Etc
### Using with pycharm
* If .bashrc file's paths don't work in pycharm environment (e.g. `nvcc -h`), open pycharm with following
```bash
# move to pycharm installed location (location may vary)
moono@moono-ubuntu:~$ cd .local/share/JetBrains/Toolbox/apps/PyCharm-P/ch-0/201.7846.77/bin/
# launch pycharm with .bashrc
moono@moono-ubuntu:~/.local/share/JetBrains/Toolbox/apps/PyCharm-P/ch-0/201.7846.77/bin$ bash pycharm.sh
```
[loss_tensorboard]: assets/tf-keras-stylegan2-loss.PNG
[generation_tensorboard]: assets/tf-keras-stylegan2-fake-images.PNG
[official_result]: assets/seed6600-official.png
[restored_result]: assets/seed6600-restored.png
[256x256_d_loss]: assets/d_loss_256x256.png
[256x256_g_loss]: assets/g_loss_256x256.png
[256x256_result_0]: assets/out_256x256_0.png
[256x256_result_1]: assets/out_256x256_1.png
[256x256_result_2]: assets/out_256x256_2.png
+216
View File
@@ -0,0 +1,216 @@
nohup python cvae.py --name="lhunor_abl_all_post_r7" > nohup/post/abl_all_post_r7.out 2>&1 &
nohup python evaluate_ae.py --name lhunor_abl_all_post_r7_70k > nohup/evaluate/abl_all_post_r7_70k.out 2>&1 &
nohup python calculate_error.py --name error_abl_all_post_r7_70k > nohup/error/abl_all_post_r7_70k.out 2>&1 &
one final convolutional layer for smoothing, on the final coordinates
skirts!!!!
pyramid next week
basic final, basic final skirts,
basic pyramid
removing curvatures test in ablation
nohup python cvae_low.py --name="lhunor_cvae_low_4_3_skirt_d32" > nohup/nohup_cvae_low_4_3_skirt_d32.out 2>&1 &
nohup python auto_encoder_body.py --name="lhunor_ae_body_post_2" > nohup/post/ae_body_2.out 2>&1 &
nohup python auto_encoder_static.py --name="lhunor_ae_static_post_4" > nohup/post/ae_static_4.out 2>&1 &
nohup python auto_encoder_curv.py --name="lhunor_ae_curv_post_1" > nohup/post/ae_curv_1.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_pyr_64_128_3" > nohup/post/abl_pyr_64_128_3.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_pyr_32_64" > nohup/post/abl_pyr_32_64.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_cvae_pyr_3_32_2_64_skirt_test_60k" > nohup/cvae_pyr_3_32_2_64_skirt_test_60k.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32" > nohup/ablation/pyr_abl_32.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_2" > nohup/ablation/pyr_abl_32_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_skin" > nohup/ablation/pyr_abl_32_skin.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_skin_offset" > nohup/ablation/pyr_abl_32_skin_offset.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_skirt" > nohup/ablation/pyr_abl_32_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_extra_loss_skirt" > nohup/ablation/pyr_abl_32_extra_loss_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_skirt_2" > nohup/ablation/pyr_abl_32_skirt_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64" > nohup/ablation/pyr_abl_32_64.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_2" > nohup/ablation/pyr_abl_32_64_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_skirt" > nohup/ablation/pyr_abl_32_64_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_skirt_2" > nohup/ablation/pyr_abl_32_64_skirt_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_cape_m" > nohup/ablation/pyr_abl_64_cape_m.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_cape_f" > nohup/ablation/pyr_abl_64_cape_f.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_cape_f_2" > nohup/ablation/pyr_abl_64_cape_f_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_128_cape_f" > nohup/ablation/pyr_abl_64_128_cape_f.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_128_cape_m" > nohup/ablation/pyr_abl_64_128_cape_m.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_128_256_cape_f_2" > nohup/ablation/pyr_abl_64_128_256_cape_f_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_128_256_512_cape_f" > nohup/ablation/pyr_abl_64_128_256_512_cape_f.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_64_128_256_512_cape_f_2" > nohup/ablation/pyr_abl_64_128_256_512_cape_f_2.out 2>&1 &
nohup python upscale_test.py --name="lhunor_256_512_skirt" > nohup/up_256_512_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_cvae_v2_7_4_3" > nohup/cvae_v2_7_4_3.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_extra_loss" > nohup/ablation/abl_extra_loss.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_extra_loss_skirt" > nohup/ablation/abl_extra_loss_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_extra_loss_skirt_2" > nohup/ablation/abl_extra_loss_skirt_2.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_curv" > nohup/ablation/abl_curv.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_curv_skirt" > nohup/ablation/abl_curv_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_abs" > nohup/ablation/abl_abs.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_abs_skirt" > nohup/ablation/abl_abs_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_skin" > nohup/ablation/abl_skin.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_skin_norm" > nohup/ablation/abl_skin_norm.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_skin_skirt" > nohup/ablation/abl_skin_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_all" > nohup/ablation/abl_all.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_all_skirt" > nohup/ablation/abl_all_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_orient" > nohup/ablation/abl_orient.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_orient_skirt" > nohup/ablation/abl_orient_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_one_hot_skirt" > nohup/ablation/abl_one_hot_skirt.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_norm" > nohup/ablation/abl_norm.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_10p" > nohup/ablation/abl_10p.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_25p" > nohup/ablation/abl_25p.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_10p_skin" > nohup/ablation/abl_10p_skin.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_25p_skin" > nohup/ablation/abl_25p_skin.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_50p_skin" > nohup/ablation/abl_50p_skin.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_skin_var" > nohup/ablation/abl_skin_var.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_skin_var_2" > nohup/ablation/abl_skin_var_2.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_skin_var_3" > nohup/ablation/abl_skin_var_3.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_skin_var_4" > nohup/ablation/abl_skin_var_4.out 2>&1 &
nohup python cvae_curv.py --name="lhunor_cvae_curv" > nohup/ablation/cvae_curv.out 2>&1 &
nohup python cvae_curv.py --name="lhunor_cvae_curv_2" > nohup/ablation/cvae_curv_2.out 2>&1 &
abl_2_pyr_32_64_128
abl_2_pyr_64_cape_f_03223_SS
abl_2_pyr_32_64_128_256_512
nohup python cvae.py --name="lhunor_cape_f_03223_SL_2" > nohup/ablation/cape_f_03223_SL_2.out 2>&1 &
nohup python cvae.py --name="lhunor_cape_f_03223_SS_2" > nohup/ablation/cape_f_03223_SS_2.out 2>&1 &
nohup python cvae.py --name="lhunor_abl_all_post" > nohup/post/abl_all_post.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32" > nohup/ablation_2/abl_2_pyr_32.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_skin" > nohup/ablation_2/abl_2_pyr_32_skin.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_skin_2" > nohup/ablation_2/abl_2_pyr_32_skin_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_skin_norm" > nohup/ablation_2/abl_2_pyr_32_skin_norm.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64" > nohup/ablation_2/abl_2_pyr_32_64.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_2" > nohup/ablation_2/abl_2_pyr_32_64_2.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_128" > nohup/ablation_2/abl_2_pyr_32_64_128.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_128_256" > nohup/ablation_2/abl_2_pyr_32_64_128_256.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_128_256_512" > nohup/ablation_2/abl_2_pyr_32_64_128_256_512.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_128_256_512_norm" > nohup/ablation_2/abl_2_pyr_32_64_128_256_512_norm.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_skirt" > nohup/ablation_2/abl_2_pyr_32_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_skirt_extra_loss" > nohup/ablation_2/abl_2_pyr_32_skirt_extra_loss.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_skirt" > nohup/ablation_2/abl_2_pyr_32_64_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_128_skirt" > nohup/ablation_2/abl_2_pyr_32_64_128_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_128_256_skirt" > nohup/ablation_2/abl_2_pyr_32_64_128_256_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_32_64_128_256_512_skirt" > nohup/ablation_2/abl_2_pyr_32_64_128_256_512_skirt.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_cape_f" > nohup/ablation_2/abl_2_pyr_64_cape_f.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_128_cape_f" > nohup/ablation_2/abl_2_pyr_64_128_cape_f.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_128_256_cape_f" > nohup/ablation_2/abl_2_pyr_64_128_256_cape_f.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_128_256_512_cape_f" > nohup/ablation_2/abl_2_pyr_64_128_256_512_cape_f.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_cape_m" > nohup/ablation_2/abl_2_pyr_64_cape_m.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_128_cape_m" > nohup/ablation_2/abl_2_pyr_64_128_cape_m.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_128_256_cape_m" > nohup/ablation_2/abl_2_pyr_64_128_256_cape_m.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_128_256_512_cape_m" > nohup/ablation_2/abl_2_pyr_64_128_256_512_cape_m.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_abl_2_pyr_64_128_256_cape_m_00032_SL" > nohup/ablation_2/abl_2_pyr_64_128_256_cape_m_00032_SL.out 2>&1 &
abl_2_all_skirt
abl_2_pyr_64_cape_m_00032_SL
python convert_to_uv.py -f '/home/workspace/data/dataset/validation' -s 0 -e 150 -n "lhunor"
python convert_to_uv.py -f '/home/workspace/data/dataset/skirt_test' -s 0 -e 1 -n "lhunor"
cp configs/config_abl_all.py configs/config_abl_abs.py
python merge_data.py -fi '/home/workspace/data/dataset/test' -fo '/home/workspace/data/dataset/test_ready' -s 0 -e 1500
python merge_data.py -fi '/home/workspace/data/dataset/valid_skirt' -fo '/home/workspace/local-data/valid_skirt' -s 0 -e 1500
nohup python upscale_test.py > nohup/u.out 2>&1 &
nohup python unpose_one_hot.py > nohup/unpose_one_hot_train_skirt.out 2>&1 &
nohup python preprocessing.py > nohup/unshape_one_hot_test_skirt.out 2>&1 &
nohup python dataset.py > nohup/avg_skirt.out 2>&1 &
nohup python evaluate_ae_support.py > nohup/evaluate_ae_support.out 2>&1 &
nohup python evaluate_cape.py --name evaluate_abl_pyr_64_128_256_512_cape_f > nohup/evaluate_abl_pyr_64_128_256_512_cape_f.out 2>&1 &
nohup python evaluate_ae_pyramid.py --name evaluate_ae_abl_pyr_32_64_110k > nohup/evaluate_ae_abl_pyr_32_64_110k.out 2>&1 &
nohup python evaluate_ae_pyramid.py --name evaluate_ae_abl_pyr_32_64_2_50k > nohup/evaluate_ae_abl_pyr_32_64_2_50k.out 2>&1 &
nohup python evaluate_ae_pyramid.py --name evaluate_ae_abl_pyr_32_skirt > nohup/evaluate_ae_abl_pyr_32_skirt.out 2>&1 &
nohup python evaluate_ae.py --name abl_norm_skirt_70k > nohup/evaluate/abl_norm_skirt_70k.out 2>&1 &
nohup python calculate_error.py --name error_abl_2_pyr_32_64_128_skirt > nohup/error_ae_abl_skin_skirt_70k.out 2>&1 &
nohup python evaluate_ae_pyramid.py --name abl_2_pyr_32_64_100k > nohup/evaluate/abl_2_pyr_32_64_100k.out 2>&1 &
nohup python evaluate_ae_pyramid.py --name abl_2_pyr_32_64_128_80k > nohup/evaluate/abl_2_pyr_32_64_128_80k.out 2>&1 &
nohup python evaluate_ae_pyramid.py --name abl_2_pyr_32_64_128_256_512_skirt_50k > nohup/evaluate/abl_2_pyr_32_64_128_256_512_skirt_50k.out 2>&1 &
nohup python calculate_error.py --name abl_2_pyr_32_64_128_256_512_50k > nohup/error/abl_2_pyr_32_64_128_256_512_50k.out 2>&1 &
nohup 7z x CAPE_dataset.zip.001 > nohup/unzip.out 2>&1 &
nohup python main.py --config configs/CAPE-affineconv_nz18_pose24_clotype8_male.yaml --name CAPE-affineconv_nz18_pose24_clotype8_male --gender male --mode demo > cape_male.out 2>&1 &
nohup python3 main.py --config configs/CAPE-affineconv_nz18_pose24_clotype8_female.yaml --name CAPE-affineconv_nz18_pose24_clotype8_female --gender female --mode demo > cape_female.out 2>&1 &
nohup python curvature_labels.py > nohup/dijkstra_skirt.out 2>&1 &
nohup python downscale.py > nohup/downscale_2_test_skirt.out 2>&1 &
nohup python downscale_cape.py > nohup/downscale_cape_train_m.out 2>&1 &
nohup python merge_data.py -fi /home/workspace/data/dataset/train -fo /home/workspace/local-data/train -s 0 -e 1500 > nohup/merge_data_valid_skirt.out 2>&1 &
nohup python merge_data.py -fi /home/workspace/data/dataset/test_skirt -fo /home/workspace/data/dataset/test_skirt_ready -s 0 -e 1500 > nohup/merge_data_test_skirt_ready.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 0 -e 100 -n lhunor_unpose_skirt_0_100 > nohup/unpose_skirt_0_100.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 100 -e 200 -n lhunor_unpose_skirt_100_200 > nohup/unpose_skirt_100_200.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 200 -e 300 -n lhunor_unpose_skirt_200_300 > nohup/unpose_skirt_200_300.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 300 -e 400 -n lhunor_unpose_skirt_300_400 > nohup/unpose_skirt_300_400.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 400 -e 500 -n lhunor_unpose_skirt_400_500 > nohup/unpose_skirt_400_500.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 500 -e 600 -n lhunor_unpose_skirt_500_600 > nohup/unpose_skirt_500_600.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 600 -e 700 -n lhunor_unpose_skirt_600_700 > nohup/unpose_skirt_600_700.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/train_skirt -s 700 -e 800 -n lhunor_unpose_skirt_700_800 > nohup/unpose_skirt_700_800.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/test_skirt -s 0 -e 100 -n lhunor_unpose_test_skirt_0_100 > nohup/unpose_test_skirt_0_100_2.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/test_skirt -s 100 -e 200 -n lhunor_unpose_test_skirt_100_200 > nohup/unpose_test_skirt_100_200_2.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/cloth3d_skirt -s 0 -e 100 -n lhunor_unpose_extra_skirt_0_100 > nohup/unpose_extra_skirt_0_100.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/cloth3d_skirt -s 100 -e 200 -n lhunor_unpose_extra_skirt_100_200 > nohup/unpose_extra_skirt_100_200.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/cloth3d_skirt -s 200 -e 300 -n lhunor_unpose_extra_skirt_200_300 > nohup/unpose_extra_skirt_200_300_2.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/cloth3d_skirt -s 300 -e 400 -n lhunor_unpose_extra_skirt_300_400 > nohup/unpose_extra_skirt_300_400.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/test -s 0 -e 100 -n lhunor_unpose_test_0_100 > nohup/unpose_test_0_100.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/test -s 100 -e 200 -n lhunor_unpose_test_100_200 > nohup/unpose_test_100_200.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/test -s 200 -e 300 -n lhunor_unpose_test_200_300 > nohup/unpose_test_200_300.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/test -s 300 -e 400 -n lhunor_unpose_test_300_400 > nohup/unpose_test_300_400.out 2>&1 &
nohup python unpose_skirt.py -f /home/workspace/data/dataset/test -s 00 -e 00 -n lhunor_unpose_test_00_00 > nohup/unpose_test_00_00.out 2>&1 &
calculate avg garment for skirt
need dijkstra/dijsktra skirt
bodies have to be converted in non-skirt mode
need to pretrain encoders again...
not enough sapce to preprocess data
add one more layer and train, repeat
curvature binning
do it according to trainingn set distribution for validation, or it's own for validation
merge curvatures for validation from data-net to local
avg garment for big datasets
check ranges of curvatur ranges
collision in pecentage
bend loss?
evaluation pipeline
look tailernot dataset
blender for visualization
gpu needed estimation
let's say if we want to finish in on eweek
we use the same skinning weights for skirts. okay?
redo curvatures for skirt train.valid
train_skirt: array([1.83511008e-06, 1.84299448e-03, 3.38148023e-03, 5.67459036e-03, 1.02263279e-02, 2.21962575e-02, 3.24614763e-01])
array([1.83511008e-06, 1.82755420e-03, 3.35242413e-03, 5.61911613e-03, 1.01151643e-02, 2.20824350e-02, 3.25994939e-01])
valid_skirt: array([4.51388405e-06, 1.90253550e-03, 3.44673195e-03, 5.73266158e-03, 1.02412868e-02, 2.22220197e-02, 3.12285781e-01])
array([4.51388405e-06, 1.88061886e-03, 3.40514188e-03, 5.64374123e-03, 1.00161384e-02, 2.18187869e-02, 3.17445695e-01])
test_skirt: array([7.21150764e-06, 1.77989923e-03, 3.26598203e-03, 5.47702867e-03, 9.79944225e-03, 2.07518619e-02, 3.23069602e-01])
how to include curvature in evaluation?
use gt, predict with vae?
docker run -d --rm -p 61208-61209:61208-61209 -e GLANCES_OPT="-w" -v /var/run/docker.sock:/var/run/docker.sock:ro --pid host nicolargo/glances:alpine-dev
+1 -1
View File
@@ -59,7 +59,7 @@ if __name__ == '__main__':
folder = config['test_dir'] folder = config['test_dir']
checkpoint_name = config['name'] checkpoint_name = config['name']
iteration = '' iteration = '70000'
F = np.load("custom_utils/smpl_faces" + skirt_modifier + ".npy").astype(np.int32) F = np.load("custom_utils/smpl_faces" + skirt_modifier + ".npy").astype(np.int32)
garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item() garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item()
+2 -2
View File
@@ -19,8 +19,8 @@ config['only_curvatures'] = False
ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False) ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False)
assert ablation_count == 1 or ablation_count == 0 assert ablation_count == 1 or ablation_count == 0
config['gpu'] = '1' config['gpu'] = '0'
config['name'] = 'al_all_post' config['name'] = 'abl_all_post_r7'
config['notes'] = '' config['notes'] = ''
config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy' #custom_data/train_no_shape_avg_uv.npy # train_no_shape_avg_uv train_no_shape_one_hot_avg_uv config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy' #custom_data/train_no_shape_avg_uv.npy # train_no_shape_avg_uv train_no_shape_one_hot_avg_uv
+2 -2
View File
@@ -73,7 +73,7 @@ def kl_loss(mean, logvar):
# 3d loss on posed !!!!!!!!!!!! # 3d loss on posed !!!!!!!!!!!!
# @tf.function @tf.function
def train_step(batch): def train_step(batch):
with tf.GradientTape() as tape: with tf.GradientTape() as tape:
@@ -476,7 +476,7 @@ def test():
if __name__ == '__main__': if __name__ == '__main__':
# init_wandb() init_wandb()
parser = argparse.ArgumentParser(description='') parser = argparse.ArgumentParser(description='')
parser.add_argument('--name', type=str) parser.add_argument('--name', type=str)
args = vars(parser.parse_args()) args = vars(parser.parse_args())
+1 -1
View File
@@ -102,7 +102,7 @@ if __name__ == '__main__':
ids = [id.split('_')[0] for id in ids] ids = [id.split('_')[0] for id in ids]
ids = set(ids) ids = set(ids)
iteration = '' iteration = '70000'
load_model(model, 'checkpoints/post/' + checkpoint_name, iteration=iteration) load_model(model, 'checkpoints/post/' + checkpoint_name, iteration=iteration)
# load_model(model, 'checkpoints/' + 'abl_all_120k') # load_model(model, 'checkpoints/' + 'abl_all_120k')
+4 -4
View File
@@ -7,10 +7,10 @@ if __name__ == '__main__':
garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item() garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item()
folder = '/home/workspace/data/results/errors' folder = '/home/workspace/data/results/errors'
checkpoint_name = 'abl_2_pyr_32_64_128_256_512' checkpoint_name = 'abl_all_post_r7'
checkpoint_name_skirt = 'abl_2_pyr_32_64_128_256_512_skirt' checkpoint_name_skirt = ''
iteration = '50000' iteration = '70000'
iteration_skirt : str = '50000' iteration_skirt : str = ''
# checkpoint_name = 'abl_skin' # checkpoint_name = 'abl_skin'
# checkpoint_name_skirt = checkpoint_name + '_skirt' # checkpoint_name_skirt = checkpoint_name + '_skirt'