test postometro pipeline

common/utils/vis.py

@@ -5,7 +5,7 @@ from mpl_toolkits.mplot3d import Axes3D
 import matplotlib.pyplot as plt
 import matplotlib as mpl
 import os
-os.environ["PYOPENGL_PLATFORM"] = "osmesa"
+os.environ["PYOPENGL_PLATFORM"] = "egl"
 import pyrender
 import trimesh
 from config import cfg

main/inference.py

@@ -16,7 +16,6 @@ from utils.inference_utils import process_mmdet_results, non_max_suppression
 from postometro_utils.smpl import SMPL
 import data.config as smpl_cfg
 from postometro import get_model
-from postometro_utils.renderer_pyrender import PyRender_Renderer
 
 class Inferer:
 
@@ -116,8 +115,6 @@ class Inferer:
             img = transform(img)
             img = img.to(cfg.device)[None,:,:,:]
 
-            self.renderer = PyRender_Renderer(resolution=(bbox[2], bbox[3]), faces=self.faces)
-
             # mesh recovery
             with torch.no_grad():
                 out = self.hmr_model(img)
@@ -134,13 +131,6 @@ class Inferer:
             cx_delta, cy_delta = bbox_cx / img_cx - 1, bbox_cy / img_cy - 1
             
             # render single person mesh
-            # focal = [self.cfg.focal[0] / self.cfg.renderer_input_body_shape[1] * bbox[2], self.cfg.focal[1] / self.cfg.renderer_input_body_shape[0] * bbox[3]]
-            # princpt = [self.cfg.princpt[0] / self.cfg.renderer_input_body_shape[1] * bbox[2] + bbox[0], self.cfg.princpt[1] / self.cfg.renderer_input_body_shape[0] * bbox[3] + bbox[1]]
-            # vis_img = render_mesh(vis_img, pred_3d_vertices_fine, self.faces, {'focal': focal, 'princpt': princpt}, 
-            #                       mesh_as_vertices=mesh_as_vertices)
-            # vis_img = render_mesh(vis_img, pred_3d_vertices_fine, self.faces, [pred_cam[0] / (original_img_width / bbox[2]), pred_cam[0] / (original_img_height / bbox[3]), pred_cam[1], pred_cam[2]], mesh_as_vertices=mesh_as_vertices)
-            # import ipdb
-            # ipdb.set_trace()
             vis_img = render_mesh(vis_img, pred_3d_vertices_fine, self.faces, [pred_cam[0] / (original_img_width / bbox[2]), pred_cam[0] / (original_img_height / bbox[3]), 
                                                                                pred_cam[1] + cx_delta / (pred_cam[0] / (original_img_width / bbox[2])), 
                                                                                pred_cam[2] + cy_delta / (pred_cam[0] / (original_img_height / bbox[3]))],