def merge_octrees(octrees, labels):
return octree_batch(octrees), labels
def merge_octrees(octrees1, octrees2, labels):
bo1 = octree_batch(octrees1)
bo2 = octree_batch(octrees2)
#tf.print(tf.shape(labels),tf.shape(bo1),tf.shape(bo2),"taking labels")
return bo1, bo2, labels
def merge_octrees(octrees, *args):
octree = octree_batch(octrees)
return (octree, ) + args
def merge_octrees(octrees0, octrees1, *args):
octree0 = octree_batch(octrees0)
octree1 = octree_batch(octrees1)
return (octree0, octree1) + args