test mmdet pipeline

app.py

@@ -14,7 +14,7 @@ except:
     os.system('pip install /home/user/app/main/transformer_utils')
 
 os.system('cp -rf /home/user/app/assets/conversions.py /home/user/.pyenv/versions/3.9.18/lib/python3.9/site-packages/torchgeometry/core/conversions.py')
-DEFAULT_MODEL='postometro'
+DEFAULT_MODEL='smpler_x_h32' # for config
 OUT_FOLDER = '/home/user/app/demo_out'
 os.makedirs(OUT_FOLDER, exist_ok=True)
 

common/utils/preprocessing.py

@@ -3,7 +3,7 @@ import cv2
 import random
 from config import cfg
 import math
-from utils.human_models import smpl_x, smpl
+# from utils.human_models import smpl_x, smpl
 from utils.transforms import cam2pixel, transform_joint_to_other_db
 from plyfile import PlyData, PlyElement
 import torch

main/inference.py

@@ -25,12 +25,13 @@ class Inferer:
         print("Infer using device: ", self.device)
         
         # # load model config
-        # config_path = osp.join(CUR_DIR, './config', f'config_{pretrained_model}.py')
+        config_path = osp.join(CUR_DIR, './config', f'config_{pretrained_model}.py')
         # ckpt_path = osp.join(CUR_DIR, '../pretrained_models', f'{pretrained_model}.pth.tar')
-        # cfg.get_config_fromfile(config_path)
-        # cfg.update_config(num_gpus, ckpt_path, output_folder, self.device)
-        # self.cfg = cfg
-        # cudnn.benchmark = True
+        ckpt_path = None # for config
+        cfg.get_config_fromfile(config_path)
+        cfg.update_config(num_gpus, ckpt_path, output_folder, self.device)
+        self.cfg = cfg
+        cudnn.benchmark = True
         
         # # load model
         # from base import Demoer
@@ -47,8 +48,8 @@ class Inferer:
 
     def infer(self, original_img, iou_thr, frame, multi_person=False, mesh_as_vertices=False):
         from utils.preprocessing import process_bbox, generate_patch_image
-        from utils.vis import render_mesh, save_obj
-        from utils.human_models import smpl_x
+        # from utils.vis import render_mesh, save_obj
+        # from utils.human_models import smpl_x
         mesh_paths = []
         smplx_paths = []
         # prepare input image