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
