def _visualize_gui_naive(self, meshList, skelList=None, body_bbox_list=None, img_original=None, normal_compute=True):
"""
args:
meshList: list of {'ver': pred_vertices, 'f': smpl.faces}
skelList: list of [JointNum*3, 1] (where 1 means num. of frames in glviewer)
bbr_list: list of [x,y,w,h]
"""
if body_bbox_list is not None:
for bbr in body_bbox_list:
viewer2D.Vis_Bbox(img_original, bbr)
# viewer2D.ImShow(img_original)
glViewer.setWindowSize(img_original.shape[1], img_original.shape[0])
# glViewer.setRenderOutputSize(inputImg.shape[1],inputImg.shape[0])
glViewer.setBackgroundTexture(img_original)
glViewer.SetOrthoCamera(True)
glViewer.setMeshData(meshList, bComputeNormal= normal_compute) # meshes = {'ver': pred_vertices, 'f': smplWrapper.f}
if skelList is not None:
glViewer.setSkeleton(skelList)
if True: #Save to File
if True: #Cam view rendering
# glViewer.setSaveFolderName(overlaidImageFolder)
glViewer.setNearPlane(50)
glViewer.setWindowSize(img_original.shape[1], img_original.shape[0])
# glViewer.show_SMPL(bSaveToFile = True, bResetSaveImgCnt = False, countImg = False, mode = 'camera')
glViewer.show(1)
if False: #Side view rendering
# glViewer.setSaveFolderName(sideImageFolder)
glViewer.setNearPlane(50)
glViewer.setWindowSize(img_original.shape[1], img_original.shape[0])
glViewer.show_SMPL(bSaveToFile = True, bResetSaveImgCnt = False, countImg = True, zoom=1108, mode = 'youtube')
def renderSMPLoutput(rootDir='/home/hjoo/temp/render_general',
rendermode='overlaid',
rendertype='mesh',
imgname='render'):
if os.path.exists(rootDir) == False:
os.mkdir(rootDir)
if rendermode == 'side':
targetFolder = rootDir + '/side'
if rendertype == 'mesh':
glViewer.g_bShowSkeleton = False
glViewer.g_bShowMesh = True
targetFolder += "_mesh"
elif rendertype == 'skeleton':
glViewer.g_bShowSkeleton = True
glViewer.g_bShowMesh = False
targetFolder += "_skel"
else:
assert (False)
if os.path.exists(targetFolder) == False:
os.mkdir(targetFolder)
glViewer.show(1)
glViewer.setSaveFolderName(targetFolder)
glViewer.setSaveImgName(imgname)
glViewer.show_SMPL(bSaveToFile=True,
bResetSaveImgCnt=False,
countImg=True,
mode='side',
zoom=1600,
bShowBG=False)
# glViewer.show(0)
elif rendermode == 'overlaid':
targetFolder = rootDir + '/overlaid'
if rendertype == 'mesh':
glViewer.g_bShowSkeleton = False
glViewer.g_bShowMesh = True
targetFolder += "_mesh"
elif rendertype == 'skeleton':
glViewer.g_bShowSkeleton = True
glViewer.g_bShowMesh = False
targetFolder += "_skel"
elif rendertype == 'raw':
glViewer.g_bShowSkeleton = False
glViewer.g_bShowMesh = False
targetFolder += "_raw"
else:
assert (False)
if os.path.exists(targetFolder) == False:
os.mkdir(targetFolder)
glViewer.setSaveFolderName(targetFolder)
glViewer.setSaveImgName(imgname)
glViewer.show_SMPL(bSaveToFile=True,
bResetSaveImgCnt=False,
countImg=True,
mode='camera')
else:
assert (False)