rendertrainingimages.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:LabelFusion 作者: RobotLocomotion 项目源码 文件源码
def captureLabelImage(self, filename):
        view = self.view
        self.disableLighting()
        im = sgp.saveScreenshot(view, filename, shouldRender=False, 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]
            print 'writing:', filename
            scipy.misc.imsave(filename, img)

        return im
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号