/RT-4K-CGH

code for RT 4K CGH

Primary LanguagePythonApache License 2.0Apache-2.0

RT-4K-CGH

1920

Real-time 4K CGH,paper:https://www.nature.com/articles/s41598-023-46575-1

environment set up

The environment is the same as https://github.com/flyingwolfz/CCNN-CGH/tree/main

conda create -n ccnncgh python=3.9
conda activate ccnncgh
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
pip install opencv-python
pip install tqdm
pip install scipy
pip install scikit-image

The complexPyTorch is from https://github.com/wavefrontshaping/complexPyTorch

The ASM is from https://github.com/computational-imaging/neural-holography

download the pretrained phases from BaiduNetdisk: https://pan.baidu.com/s/1yXb8KisjLYeLGKCJaljMjQ?pwd=q3fw

or google drive: https://drive.google.com/drive/folders/1462bOb1uWxweCejxQ5OBxWYLljMMDiAu?usp=sharing