def captureColorImage(self, filename): view = self.view self.enableLighting() print 'writing:', filename im = sgp.saveScreenshot(view, filename, shouldRender=False, shouldWrite=True) return im
def captureLabelImage(self, filename): view = self.globalsDict['view'] self.disableLighting() im = sgp.saveScreenshot(view, filename, shouldWrite=False) if filename is not None: img = vnp.getNumpyFromVtk(im, 'ImageScalars') assert img.dtype == np.uint8 img.shape = (im.GetDimensions()[1], im.GetDimensions()[0], 3) img = np.flipud(img) img = img[:, :, 0] scipy.misc.imsave(filename, img) return im
def captureColorImage(self, filename): view = self.globalsDict['view'] self.enableLighting() print 'writing:', filename im = sgp.saveScreenshot(view, filename, shouldWrite=True) return im