d2nn
There are 1 repositories under d2nn topic.
feng-spinning/onn-simulation
A program using pytorch to simulate diffracive deep nerual network (D2NN), containing updates (relative to the original paper) of incoherent propagation, parallel computation, nonlinear activation, changing of propagating distance.