Deep Belief Networks for image reconstruction in Tensorflow 2.
DBN/crbm.py
file contains the class where all logic and training code was implemented
The code for CRBM was adapted from this repository. However, it's not being used the max pooling output, since we want to keep the spatial dimensions of the images.
DBN/RBM/rbm.py
file contains the class where all logic and training code was implemented
Is not being used and the codes needs to be adapted for the new implementation of DBN