lxm0503 / phase_retrieval_algorithms

Pytorch implementation for Error Reduction algorithm and Hybrid input-output algorithm.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

phase_retrieval_algorithms

Pytorch implementation of phase retrieval algorithms. Currently, Error reduction & Hybrid input-output algorithms are implemented.

Requirements

  • pytorch
  • torchvision
  • numpy
  • PIL
  • tqdm

Usage

Required argument

--algorithm         ER or HIO

--num_iterations    Number of iterations for each algorithm.

--num_repeats       Number of repeats for each sample.

--data_root         Data directory

--batch_size        Size of mini-batch

--gpu               Index for gpu device.

About

Pytorch implementation for Error Reduction algorithm and Hybrid input-output algorithm.


Languages

Language:Python 99.4%Language:Shell 0.6%