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)
