Reshape is correct?
manuelsh opened this issue · comments
Manuel commented
CapsNet-Tensorflow/capsLayer.py
Line 78 in 4be551a
I think this line is not preserving the following in the paper:
"Each primary capsule output sees the outputs of all 256 × 81 Conv1 units whose receptive fields overlap with the location of the center of the capsule."
i.e. we should ensure that the first capsule after the view corresponds to the pixel [0,0] of the first 8 filters, and the second with [0,1] and so on.
Manuel commented
I believe the reshape is ok. No issue.