sspku-2021 / PBCNN

15 stars 8 forks source link

In encoder(def) of enhanced_pbcnn.py, maybe the convolutional layer should be 8-8-16-16-32-32? #2

Open fightingha opened 2 years ago

fightingha commented 2 years ago

According to the paper, it should be conv(8)-conv(8)-pool(2)- conv(16)-conv(16)-pool(2)- conv(32)-conv(32)-pool(8)[ 8-8-16-16-32-32]. The code is 8-16-32, and each layer may be missing a convolution kernel. Perhaps the wrong version of the code is transmitted, or the simplified version of the network can achieve similar results.