输入
稀疏卷积的输入包括两部分,一个是坐标,另一个是特征。
self.scn_input = scn.InputLayer(3, sparse_shape.tolist()) # [h,w,l]
coors = coors.int()[:, [1, 2, 3, 0]] # [h, w, l, batch] 将 batch_size调换到最后一个位置
ret = self.scn_input((coors.cpu(), voxel_features, batch_size)) #[45287, 4] [45287, 1] 2
稀疏卷积的spatial_size确定:
self.spatial_size = self.sparseModel.input_spatial_size(torch.LongTensor([1,1]))
3d稀疏卷积
m = None
m = scn.Sequential()
m.add(scn.SubmanifoldConvolution(3, i, o, 3, False))
m.add(scn.BatchNormLeakyReLU(i ,leakiness=leakiness)).add(
scn.Convolution(dimension, i, o, 3, 2, False))
def block(self, m, a, b, dimension=3, residual_blocks=False, leakiness=0): # default using residual_block
if residual_blocks: #ResNet style blocks
m.add(scn.ConcatTable()
.add(scn.Identity() if a == b else scn.NetworkInNetwork(a, b, False))
.add(scn.Sequential()
.add(scn.BatchNormLeakyReLU(a,leakiness=leakiness))
.add(scn.SubmanifoldConvolution(dimension, a, b, 3, False))
.add(scn.BatchNormLeakyReLU(b,leakiness=leakiness))
.add(scn.SubmanifoldConvolution(dimension, b, b, 3, False)))
).add(scn.AddTable())
else: #VGG style blocks
m.add(scn.Sequential()
.add(scn.BatchNormLeakyReLU(a,leakiness=leakiness))
.add(scn.SubmanifoldConvolution(dimension, a, b, 3, False)))
稀疏3d卷积
子流形卷积scn.SubmanifoldConvolution(2, 1, 8, 3, False): # batch_size, channel_in, channel_out, filter_size
稀疏卷积 scn.Convolution(2, 32, 32, 3, 2, False): # batch_size, channel_in, channel_out, filter_size, strides
批归一化scn.BatchNormalization(32): # channel_in
稀疏到稠密转化scn.SparseToDense(2, 32): # batch_size, channel_in
输入scn.InputBatch(2, sparse_shape): # batch_size, sparse_shape
设置输入input.set_locations(locations, features,0) # locations [N, 2] features [N, 1] 保存N个特征的位置,N个特征
输入scn.InputLayer(2, spatial_size, 2) # dimension, spatial_size, mode=3
mode == 0 if the input is guaranteed to have no duplicates
mode == 1 to use the last item at each spatial location
mode == 2 to keep the first item at each spatial location
mode == 3 to sum feature vectors sharing one spatial location
mode == 4 to average feature vectors at each spatial location
self.scn_input((coors.cpu(), voxel_features, batch_size)): 将坐标及相应的feature转换成可以处理用稀疏3d卷积处理的数据类型
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:稀疏3d卷积 - Python技术站