def getRenderer(ren_type='geo'): """ Choose renderer type geo: phong-shading denspose: densepose IUV normal: normal map torch3d: via pytorch3d TODO """ if ren_type=='geo': renderer = meshRenderer.meshRenderer() renderer.setRenderMode('geo') elif ren_type=='normal': renderer = meshRenderer.meshRenderer() renderer.setRenderMode('normal') elif ren_type=='densepose': renderer = denseposeRenderer.denseposeRenderer() elif ren_type=='torch3d': renderer = torch3dRenderer.torch3dRenderer() else: assert False renderer.offscreenMode(True) renderer.bAntiAliasing= False return renderer
def getRenderer(ren_type='geo'): """ Choose renderer type geo: phong-shading (silver color) colorgeo: phong-shading with color (need color infor. Default silver color) denspose: densepose IUV normal: normal map torch3d: via pytorch3d TODO """ if ren_type=='geo': renderer = meshRenderer.meshRenderer() renderer.setRenderMode('geo') elif ren_type=='colorgeo': renderer = meshRenderer.meshRenderer() renderer.setRenderMode('colorgeo') elif ren_type=='normal': renderer = meshRenderer.meshRenderer() renderer.setRenderMode('normal') elif ren_type=='densepose': renderer = denseposeRenderer.denseposeRenderer() elif ren_type=='torch3d': renderer = torch3dRenderer.torch3dRenderer() else: assert False renderer.offscreenMode(True) # renderer.bAntiAliasing= False return renderer
imgDir = args.img_dir smplModelDir = args.smpl_dir outputFolder = os.path.basename(inputDir) + '_dpOut' outputFolder = os.path.join('/run/media/hjoo/disk/data/eftout/', outputFolder) # # devfair # inputDir = './eft/11-08_coco_with8143' # inputDir = '/private/home/hjoo/spinOut/01-23_coco_8143_annotId' # imgDir = '/private/home/hjoo/data/coco/train2014' smpl = SMPL(smplModelDir, batch_size=1, create_transl=False) fileList = listdir(inputDir) #Check all fitting files render = denseposeRenderer.denseposeRenderer() render.offscreenMode(True) render.bAntiAliasing = False print(">> Found {} files in the fitting folder {}".format( len(fileList), inputDir)) totalCnt = 0 erroneousCnt = 0 for idx, f in enumerate(sorted(fileList)): if "_init" in f: continue #Load fileFullPath = join(inputDir, f) with open(fileFullPath, 'rb') as f: data = pickle.load(f)