MinkowskiEngine基准测试

MinkowskiEngine基准测试

介绍卷积层和小型U网络的前馈和后馈通过时间。可以将具有相同张量步幅，步幅和内核偏移的内核映射重新用于其他层，可以在大型nueral网络中使用的所有层上，分摊此页面中实验报告的时间。

使用Titan X进行实验。

实验设置

对于单卷积层实验，使用以下设置。

import MinkowskiEngine as ME

conv = ME.MinkowskiConvolution(

in_channels=3,

out_channels=32,

kernel_size=7,

stride=1,

dilation=1,

has_bias=False,

dimension=3)

使用体素大小为5cm的ScanNet测试仪进行实验。由于SparseConvNet和MinkowskiEngine使用不同的体素化算法，因此每个引擎处理的点数也有所不同。平均而言，在100个ScanNet测试室中，SparseConvNet产生25757.01点，而MinkowskiEngine产生26097.58点。

单卷积层

测试具有不同内核大小的同一个卷积层。报告每种算法处理SparseConvNet的平均时间（以秒为单位），平均为25757.011点，而MinkowskiEngine的平均时间为26097.58。

简单的UNet

net = nn.Sequential(

ME.MinkowskiConvolution(

in_channels=3,

out_channels=32,

kernel_size=5,

stride=1,

dilation=1,

has_bias=False,

dimension=3),

ME.MinkowskiConvolution(

in_channels=32,

out_channels=32,

kernel_size=2,

stride=2,

dilation=1,

has_bias=False,

dimension=3),

ME.MinkowskiConvolutionTranspose

in_channels=32,

out_channels=32,

kernel_size=2,

stride=2,

dilation=1,

has_bias=False,

dimension=3))

对于本实验，仅更改第一卷积层的内核大小。