Ejemplo n.º 1
0
kg0 = ME.KernelGenerator(
		# **kgargs
        kernel_size=2,
        stride=1,
        dilation=2,
        region_type=ME.RegionType.CUSTOM,
        region_offsets=ro0,
        dimension=3
        )

conv = ME.MinkowskiChannelwiseConvolution(
        in_channels=1,
        kernel_size=2,
        stride=1,
        dilation=1,
        bias=False,
        kernel_generator=kg0,
        dimension=3
        )

print(conv)

# import ipdb; ipdb.set_trace()

cm0 = x1.coordinate_manager
cmk0 = x1.coordinate_map_key
kmap0 = cm0.get_kernel_map(
        cmk0,
        cmk0,
        **kgargs