octree-nn / ocnn-pytorch

Octree-based 3D Convolutional Neural Networks
MIT License
156 stars 17 forks source link

Regarding downsampling layers in ounet and unet #24

Closed raoshashank closed 1 year ago

raoshashank commented 1 year ago

I notice that the downsampling layers in the ounet and unet models use ConvBnRelu as the downsamplers rather than pooling, which is typically used in such dense prediction/reconstruction networks to extract hierarchical features. Is there some relation between the two operations due to the structure of the octree itself?

wang-ps commented 1 year ago

The stride of ConvBnRelu is 2 for downsampling; a pooling layer with a stride of 2 can also be used for downsampling. The ConvBnRelu has trainable parameters, it can result in slightly better preformance than pooling.