visualize.py 文件源码

def old_viz(): 

    objects = np.load(sys.argv[1])
    if len(objects.shape)==3: 
        objects = [objects]
    for voxels in objects:
        print voxels.shape

        if connect > 0: 
            voxels_keep = (voxels >= threshold)
            voxels_keep = max_connected(voxels_keep, connect)
            voxels[np.logical_not(voxels_keep)] = 0
        if downsample_factor > 1:
            print "==> Performing downsample: factor: "+str(downsample_factor)+" method: "+downsample_method,
            voxels = downsample(voxels, downsample_factor, method=downsample_method)
        print "Done"
        visualization(voxels, threshold, title=str(ind+1), uniform_size=uniform_size, use_colormap=use_colormap)