├── .gitignore ├── Data └── data_prepare.py ├── Demo ├── PixhtLab │ ├── Demo.ipynb │ ├── Examples │ │ ├── 009_depth.npy │ │ ├── 009_depth_valid_mask.npy │ │ ├── 009_pixht_new.npy │ │ ├── 009_rgb.png │ │ ├── 009_visualization.png │ │ ├── 010_depth.npy │ │ ├── 010_depth_valid_mask.npy │ │ ├── 010_pixht_new.npy │ │ ├── 010_rgb.png │ │ ├── 010_visualization.png │ │ ├── 011_depth.npy │ │ ├── 011_depth_valid_mask.npy │ │ ├── 011_pixht_new.npy │ │ ├── 011_rgb.png │ │ ├── 011_visualization.png │ │ └── c82c09fc01e84282bc8870c263dcf81b_bg.jpg │ ├── GSSN │ │ ├── __init__.py │ │ └── inference_shadow.py │ ├── ShadowStyle │ │ ├── __init__.py │ │ └── inference │ │ │ ├── __init__.py │ │ │ ├── inference_shadow.py │ │ │ ├── params.py │ │ │ ├── ssn │ │ │ ├── __init__.py │ │ │ ├── perturb_touch.py │ │ │ ├── random_pattern.py │ │ │ ├── ssn.py │ │ │ ├── ssn_dataset.py │ │ │ ├── ssn_submodule.py │ │ │ ├── test.py │ │ │ └── test_dataset.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── html.py │ │ │ ├── html_server.py │ │ │ ├── imgs │ │ │ ├── index.html │ │ │ ├── make_html.py │ │ │ ├── net_utils.py │ │ │ ├── tensorboard_utils.py │ │ │ ├── time_utils.py │ │ │ ├── utils_file.py │ │ │ ├── vis_test_results.py │ │ │ └── visdom_utils.py │ ├── Torch_Render │ │ ├── hshadow_cuda.cpp │ │ ├── hshadow_cuda_kernel.cu │ │ ├── plane_visualize.cpp │ │ ├── plane_visualize_cuda.cu │ │ ├── setup.py │ │ ├── test_ground.py │ │ └── test_hshadow.py │ ├── camera.py │ ├── gssn_demo.py │ ├── hshadow_render.py │ ├── reflect_render.py │ ├── utils.py │ └── xyh_proj.py └── SSN │ ├── app.py │ ├── configs │ └── SSN.yaml │ ├── imgs │ ├── bike.png │ ├── cloud.png │ ├── flower_mask.png │ ├── human2.png │ ├── imgs_convert_rgb_alpha.py │ ├── imgs_test.py │ ├── man.png │ ├── plant1.png │ └── woman.png │ ├── model_utils.py │ ├── models │ ├── SSN.py │ ├── SSN_Model.py │ ├── abs_model.py │ └── blocks.py │ └── weights │ └── .placeholder ├── Docker ├── Dockerfile ├── build_docker.sh ├── imagen │ ├── .gitignore │ ├── .gitmodules │ ├── .travis.yml │ ├── LICENSE.txt │ ├── MANIFEST.in │ ├── README.rst │ ├── build │ │ └── lib │ │ │ └── imagen │ │ │ ├── __init__.py │ │ │ ├── audio.py │ │ │ ├── colorspaces.py │ │ │ ├── deprecated.py │ │ │ ├── image.py │ │ │ ├── patterncoordinator.py │ │ │ ├── patternfn.py │ │ │ ├── patterngenerator.py │ │ │ ├── random.py │ │ │ └── transferfn │ │ │ ├── __init__.py │ │ │ └── sheet_tf.py │ ├── doc │ │ ├── Makefile │ │ ├── Tutorials │ │ │ ├── hegde_cc07.ipynb │ │ │ └── index.rst │ │ ├── _static │ │ │ └── patterntypes_small.png │ │ ├── _templates │ │ │ └── page.html │ │ ├── conf.py │ │ ├── images │ │ │ └── ellen_arthur.pgm │ │ ├── index.ipynb │ │ ├── index.rst │ │ ├── notebook.json │ │ └── site_map.rst │ ├── imagen │ │ ├── __init__.py │ │ ├── audio.py │ │ ├── colorspaces.py │ │ ├── deprecated.py │ │ ├── image.py │ │ ├── patterncoordinator.py │ │ ├── patternfn.py │ │ ├── patterngenerator.py │ │ ├── random.py │ │ └── transferfn │ │ │ ├── __init__.py │ │ │ └── sheet_tf.py │ ├── images │ │ └── ellen_arthur.pgm │ ├── meta.yaml │ ├── setup.py │ └── tests │ │ ├── __init__.py │ │ ├── test_line.py │ │ ├── testpatterngenerator.py │ │ └── testtimedependentrandom.py ├── pixht.def └── zsh-in-docker.sh ├── Figs ├── SSG.gif ├── controllable_reflection.png ├── more_results2.png └── ssn_teaser.png ├── README.md ├── Train ├── app │ ├── Evaluater.py │ ├── Trainer.py │ ├── __init__.py │ ├── datasets │ │ ├── GSSN_Dataset.py │ │ ├── GSSN_Testing_Dataset.py │ │ ├── SSN_Dataset.py │ │ ├── __init__.py │ │ ├── camera.py │ │ └── test_dataset.py │ ├── eval │ │ ├── evaluate.py │ │ ├── utils.py │ │ └── visualize.py │ ├── metrics │ │ ├── __init__.py │ │ ├── metric.py │ │ └── pytorch_ssim │ │ │ └── __init__.py │ ├── models │ │ ├── Attention.ipynb │ │ ├── Attention_SSN.py │ │ ├── Attention_Unet.py │ │ ├── GSSN.py │ │ ├── Loss │ │ │ ├── Loss.py │ │ │ ├── __init__.py │ │ │ ├── pytorch_ssim │ │ │ │ └── __init__.py │ │ │ └── vgg19_loss.py │ │ ├── SSN.py │ │ ├── SSN_Model.py │ │ ├── SSN_v1.py │ │ ├── Sparse_PH.py │ │ ├── __init__.py │ │ ├── abs_model.py │ │ ├── attention.py │ │ ├── blocks.py │ │ ├── pvt_attention.py │ │ └── template.py │ └── utils │ │ ├── __init__.py │ │ ├── html_maker.py │ │ ├── utils.py │ │ └── vis_writer.py └── configs │ ├── GSSN.yaml │ ├── SSN.yaml │ ├── SSN_gelu.yaml │ └── SSN_gelu_1e-3.yaml └── env.sh /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | .idea 3 | *.log 4 | tmp/ 5 | *__pycache__* 6 | -------------------------------------------------------------------------------- /Data/data_prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Data/data_prepare.py -------------------------------------------------------------------------------- /Demo/PixhtLab/Demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Demo.ipynb -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/009_depth.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/009_depth.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/009_depth_valid_mask.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/009_depth_valid_mask.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/009_pixht_new.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/009_pixht_new.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/009_rgb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/009_rgb.png -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/009_visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/009_visualization.png -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/010_depth.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/010_depth.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/010_depth_valid_mask.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/010_depth_valid_mask.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/010_pixht_new.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/010_pixht_new.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/010_rgb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/010_rgb.png -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/010_visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/010_visualization.png -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/011_depth.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/011_depth.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/011_depth_valid_mask.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/011_depth_valid_mask.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/011_pixht_new.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/011_pixht_new.npy -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/011_rgb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/011_rgb.png -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/011_visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/011_visualization.png -------------------------------------------------------------------------------- /Demo/PixhtLab/Examples/c82c09fc01e84282bc8870c263dcf81b_bg.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Examples/c82c09fc01e84282bc8870c263dcf81b_bg.jpg -------------------------------------------------------------------------------- /Demo/PixhtLab/GSSN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Demo/PixhtLab/GSSN/inference_shadow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/GSSN/inference_shadow.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/inference_shadow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/inference_shadow.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/params.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/perturb_touch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/ssn/perturb_touch.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/random_pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/ssn/random_pattern.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/ssn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/ssn/ssn.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/ssn_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/ssn/ssn_dataset.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/ssn_submodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/ssn/ssn_submodule.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/ssn/test.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/ssn/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/ssn/test_dataset.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/html.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/html_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/html_server.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/imgs: -------------------------------------------------------------------------------- 1 | /home/ysheng/Dataset/dropbox/Research/SSN_Training_Share/touch -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/index.html -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/make_html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/make_html.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/net_utils.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/tensorboard_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/tensorboard_utils.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/time_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/time_utils.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/utils_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/utils_file.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/vis_test_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/vis_test_results.py -------------------------------------------------------------------------------- /Demo/PixhtLab/ShadowStyle/inference/utils/visdom_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/ShadowStyle/inference/utils/visdom_utils.py -------------------------------------------------------------------------------- /Demo/PixhtLab/Torch_Render/hshadow_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Torch_Render/hshadow_cuda.cpp -------------------------------------------------------------------------------- /Demo/PixhtLab/Torch_Render/hshadow_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Torch_Render/hshadow_cuda_kernel.cu -------------------------------------------------------------------------------- /Demo/PixhtLab/Torch_Render/plane_visualize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Torch_Render/plane_visualize.cpp -------------------------------------------------------------------------------- /Demo/PixhtLab/Torch_Render/plane_visualize_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Torch_Render/plane_visualize_cuda.cu -------------------------------------------------------------------------------- /Demo/PixhtLab/Torch_Render/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Torch_Render/setup.py -------------------------------------------------------------------------------- /Demo/PixhtLab/Torch_Render/test_ground.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Torch_Render/test_ground.py -------------------------------------------------------------------------------- /Demo/PixhtLab/Torch_Render/test_hshadow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/Torch_Render/test_hshadow.py -------------------------------------------------------------------------------- /Demo/PixhtLab/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/camera.py -------------------------------------------------------------------------------- /Demo/PixhtLab/gssn_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/gssn_demo.py -------------------------------------------------------------------------------- /Demo/PixhtLab/hshadow_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/hshadow_render.py -------------------------------------------------------------------------------- /Demo/PixhtLab/reflect_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/reflect_render.py -------------------------------------------------------------------------------- /Demo/PixhtLab/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/utils.py -------------------------------------------------------------------------------- /Demo/PixhtLab/xyh_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/PixhtLab/xyh_proj.py -------------------------------------------------------------------------------- /Demo/SSN/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/app.py -------------------------------------------------------------------------------- /Demo/SSN/configs/SSN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/configs/SSN.yaml -------------------------------------------------------------------------------- /Demo/SSN/imgs/bike.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/bike.png -------------------------------------------------------------------------------- /Demo/SSN/imgs/cloud.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/cloud.png -------------------------------------------------------------------------------- /Demo/SSN/imgs/flower_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/flower_mask.png -------------------------------------------------------------------------------- /Demo/SSN/imgs/human2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/human2.png -------------------------------------------------------------------------------- /Demo/SSN/imgs/imgs_convert_rgb_alpha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/imgs_convert_rgb_alpha.py -------------------------------------------------------------------------------- /Demo/SSN/imgs/imgs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/imgs_test.py -------------------------------------------------------------------------------- /Demo/SSN/imgs/man.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/man.png -------------------------------------------------------------------------------- /Demo/SSN/imgs/plant1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/plant1.png -------------------------------------------------------------------------------- /Demo/SSN/imgs/woman.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/imgs/woman.png -------------------------------------------------------------------------------- /Demo/SSN/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/model_utils.py -------------------------------------------------------------------------------- /Demo/SSN/models/SSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/models/SSN.py -------------------------------------------------------------------------------- /Demo/SSN/models/SSN_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/models/SSN_Model.py -------------------------------------------------------------------------------- /Demo/SSN/models/abs_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/models/abs_model.py -------------------------------------------------------------------------------- /Demo/SSN/models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Demo/SSN/models/blocks.py -------------------------------------------------------------------------------- /Demo/SSN/weights/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/Dockerfile -------------------------------------------------------------------------------- /Docker/build_docker.sh: -------------------------------------------------------------------------------- 1 | docker build -t yichensheng/pixht . -------------------------------------------------------------------------------- /Docker/imagen/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/.gitignore -------------------------------------------------------------------------------- /Docker/imagen/.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/.gitmodules -------------------------------------------------------------------------------- /Docker/imagen/.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/.travis.yml -------------------------------------------------------------------------------- /Docker/imagen/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/LICENSE.txt -------------------------------------------------------------------------------- /Docker/imagen/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/MANIFEST.in -------------------------------------------------------------------------------- /Docker/imagen/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/README.rst -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/__init__.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/audio.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/colorspaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/colorspaces.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/deprecated.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/image.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/patterncoordinator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/patterncoordinator.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/patternfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/patternfn.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/patterngenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/patterngenerator.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/random.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/transferfn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/transferfn/__init__.py -------------------------------------------------------------------------------- /Docker/imagen/build/lib/imagen/transferfn/sheet_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/build/lib/imagen/transferfn/sheet_tf.py -------------------------------------------------------------------------------- /Docker/imagen/doc/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/Makefile -------------------------------------------------------------------------------- /Docker/imagen/doc/Tutorials/hegde_cc07.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/Tutorials/hegde_cc07.ipynb -------------------------------------------------------------------------------- /Docker/imagen/doc/Tutorials/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/Tutorials/index.rst -------------------------------------------------------------------------------- /Docker/imagen/doc/_static/patterntypes_small.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/_static/patterntypes_small.png -------------------------------------------------------------------------------- /Docker/imagen/doc/_templates/page.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/_templates/page.html -------------------------------------------------------------------------------- /Docker/imagen/doc/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/conf.py -------------------------------------------------------------------------------- /Docker/imagen/doc/images/ellen_arthur.pgm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/images/ellen_arthur.pgm -------------------------------------------------------------------------------- /Docker/imagen/doc/index.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/index.ipynb -------------------------------------------------------------------------------- /Docker/imagen/doc/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/index.rst -------------------------------------------------------------------------------- /Docker/imagen/doc/notebook.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/notebook.json -------------------------------------------------------------------------------- /Docker/imagen/doc/site_map.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/doc/site_map.rst -------------------------------------------------------------------------------- /Docker/imagen/imagen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/__init__.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/audio.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/colorspaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/colorspaces.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/deprecated.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/image.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/patterncoordinator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/patterncoordinator.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/patternfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/patternfn.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/patterngenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/patterngenerator.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/random.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/transferfn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/transferfn/__init__.py -------------------------------------------------------------------------------- /Docker/imagen/imagen/transferfn/sheet_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/imagen/transferfn/sheet_tf.py -------------------------------------------------------------------------------- /Docker/imagen/images/ellen_arthur.pgm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/images/ellen_arthur.pgm -------------------------------------------------------------------------------- /Docker/imagen/meta.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/meta.yaml -------------------------------------------------------------------------------- /Docker/imagen/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/setup.py -------------------------------------------------------------------------------- /Docker/imagen/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Docker/imagen/tests/test_line.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/tests/test_line.py -------------------------------------------------------------------------------- /Docker/imagen/tests/testpatterngenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/tests/testpatterngenerator.py -------------------------------------------------------------------------------- /Docker/imagen/tests/testtimedependentrandom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/imagen/tests/testtimedependentrandom.py -------------------------------------------------------------------------------- /Docker/pixht.def: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/pixht.def -------------------------------------------------------------------------------- /Docker/zsh-in-docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Docker/zsh-in-docker.sh -------------------------------------------------------------------------------- /Figs/SSG.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Figs/SSG.gif -------------------------------------------------------------------------------- /Figs/controllable_reflection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Figs/controllable_reflection.png -------------------------------------------------------------------------------- /Figs/more_results2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Figs/more_results2.png -------------------------------------------------------------------------------- /Figs/ssn_teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Figs/ssn_teaser.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/README.md -------------------------------------------------------------------------------- /Train/app/Evaluater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/Evaluater.py -------------------------------------------------------------------------------- /Train/app/Trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/Trainer.py -------------------------------------------------------------------------------- /Train/app/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Train/app/datasets/GSSN_Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/datasets/GSSN_Dataset.py -------------------------------------------------------------------------------- /Train/app/datasets/GSSN_Testing_Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/datasets/GSSN_Testing_Dataset.py -------------------------------------------------------------------------------- /Train/app/datasets/SSN_Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/datasets/SSN_Dataset.py -------------------------------------------------------------------------------- /Train/app/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/datasets/__init__.py -------------------------------------------------------------------------------- /Train/app/datasets/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/datasets/camera.py -------------------------------------------------------------------------------- /Train/app/datasets/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/datasets/test_dataset.py -------------------------------------------------------------------------------- /Train/app/eval/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/eval/evaluate.py -------------------------------------------------------------------------------- /Train/app/eval/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/eval/utils.py -------------------------------------------------------------------------------- /Train/app/eval/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/eval/visualize.py -------------------------------------------------------------------------------- /Train/app/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Train/app/metrics/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/metrics/metric.py -------------------------------------------------------------------------------- /Train/app/metrics/pytorch_ssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/metrics/pytorch_ssim/__init__.py -------------------------------------------------------------------------------- /Train/app/models/Attention.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/Attention.ipynb -------------------------------------------------------------------------------- /Train/app/models/Attention_SSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/Attention_SSN.py -------------------------------------------------------------------------------- /Train/app/models/Attention_Unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/Attention_Unet.py -------------------------------------------------------------------------------- /Train/app/models/GSSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/GSSN.py -------------------------------------------------------------------------------- /Train/app/models/Loss/Loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/Loss/Loss.py -------------------------------------------------------------------------------- /Train/app/models/Loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Train/app/models/Loss/pytorch_ssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/Loss/pytorch_ssim/__init__.py -------------------------------------------------------------------------------- /Train/app/models/Loss/vgg19_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/Loss/vgg19_loss.py -------------------------------------------------------------------------------- /Train/app/models/SSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/SSN.py -------------------------------------------------------------------------------- /Train/app/models/SSN_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/SSN_Model.py -------------------------------------------------------------------------------- /Train/app/models/SSN_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/SSN_v1.py -------------------------------------------------------------------------------- /Train/app/models/Sparse_PH.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/Sparse_PH.py -------------------------------------------------------------------------------- /Train/app/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/__init__.py -------------------------------------------------------------------------------- /Train/app/models/abs_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/abs_model.py -------------------------------------------------------------------------------- /Train/app/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/attention.py -------------------------------------------------------------------------------- /Train/app/models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/blocks.py -------------------------------------------------------------------------------- /Train/app/models/pvt_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/pvt_attention.py -------------------------------------------------------------------------------- /Train/app/models/template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/models/template.py -------------------------------------------------------------------------------- /Train/app/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Train/app/utils/html_maker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/utils/html_maker.py -------------------------------------------------------------------------------- /Train/app/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/utils/utils.py -------------------------------------------------------------------------------- /Train/app/utils/vis_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/app/utils/vis_writer.py -------------------------------------------------------------------------------- /Train/configs/GSSN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/configs/GSSN.yaml -------------------------------------------------------------------------------- /Train/configs/SSN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/configs/SSN.yaml -------------------------------------------------------------------------------- /Train/configs/SSN_gelu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/configs/SSN_gelu.yaml -------------------------------------------------------------------------------- /Train/configs/SSN_gelu_1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/Train/configs/SSN_gelu_1e-3.yaml -------------------------------------------------------------------------------- /env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShengCN/PixHtLab-Src/HEAD/env.sh --------------------------------------------------------------------------------