search

kumar-shridhar / PyTorch-BayesianCNN

Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.
MIT License
1.42k stars 323 forks source link

There is a problem with applying BayesianCNN to image restoration #21

Closed GGyan closed 4 years ago

GGyan commented 5 years ago

Hello, I am trying to apply the latest program in the direction of image restoration. The current results do not see the noise problem. However, during the test, I found that the image I wanted was not displayed well. The color of the image has changed and the image details and features have disappeared. Because I simply nested the BBBConv2d related program, the activation function used is relu. I don't know if it is wrong for me. If I change BBBConv2d to nn.Conv2d, my program can get the desired result. Here are the results I got: guogong1 Original image： 1gugong

kumar-shridhar commented 4 years ago

please check with this version. Thanks