def grab_images(): im = DS.getColourMap() cv2.imwrite("%s/rgb.png"%(imdir), im) im = DS.getDepthMap() cv2.imwrite("%s/depth.png"%(imdir), im) im = DS.getConfidenceMap() cv2.imwrite("%s/confidence.png"%(imdir), im) im = DS.getDepthColouredMap() cv2.imwrite("%s/rgbd.png"%(imdir), im) im = DS.getGreyScaleMap() cv2.imwrite("%s/gscale.png"%(imdir), im) im = DS.getSyncMap() np.save("%s/sync"%(imdir), im) im = DS.getUVMap() np.save("%s/uv"%(imdir), im) im = DS.getVertices() np.save("%s/vert"%(imdir), im) return
def grab_images(imdir, i): im = DS.getColourMap() cv2.imwrite("%s/rgb-%03d.png"%(imdir,i), im) im = DS.getDepthMap() cv2.imwrite("%s/depth-%03d.png"%(imdir,i), im) im = DS.getConfidenceMap() cv2.imwrite("%s/confidence-%03d.png"%(imdir,i), im) im = DS.getDepthColouredMap() cv2.imwrite("%s/rgbd-%03d.png"%(imdir,i), im) im = DS.getGreyScaleMap() cv2.imwrite("%s/gscale-%03d.png"%(imdir,i), im) im = DS.getSyncMap() print im.shape np.save("%s/sync-%03d"%(imdir,i), im) im = DS.getUVMap() np.save("%s/uv-%03d"%(imdir,i), im) im = DS.getVertices() np.save("%s/vert-%03d"%(imdir,i), im) #subprocess.call(['python', dir_path + '/dsgrab.py']) """ {"href": "static/img/confidence.png", "name": "Confidence levels (image)"}, {"href": "static/img/rgbd.png", "name": "Coloured depth image"}, {"href": "static/img/gscale.png", "name": "Grey-scale image"} """ return [{"href": imdir+"/rgb-%03d.png"%i, "name": "RGB image"}, {"href": imdir+"/depth-%03d.png"%i, "name": "Depth image"}, {"href": imdir+"/confidence-%03d.png"%i, "name": "Confidence levels (image)"}, {"href": imdir+"/rgbd-%03d.png"%i, "name": "Coloured depth image"}, {"href": imdir+"/gscale-%03d.png"%i, "name": "Grey-scale image"}]
def getDepthColoured(self): ''' Return a simple cv compatiable 8bit colour image ''' depthc = ds.getDepthColouredMap() #depthc = depthc[:,:,::-1] return Image(depthc.transpose([1,0,2]))