upload smplest_x_h40 model and config

config_base.py

@@ -0,0 +1,104 @@
+config = {
+  "data": {
+    "use_cache": True,
+    "data_dir": "./data",
+    "trainset_humandata": [
+          'SynHand'],
+    "testset": 'EHF',
+    "BEDLAM_train_sample_interval": 5,
+    "SynBody_train_sample_interval": 10,
+    "EgoBody_Kinect_train_sample_interval": 10,
+    "UBody_train_sample_interval": 10,
+    "MPI_INF_3DHP_train_sample_interval": 5,
+    "InstaVariety_train_sample_interval": 10,
+    "RenBody_HiRes_train_sample_interval": 5,
+    "ARCTIC_train_sample_interval": 10,
+    "RenBody_train_sample_interval": 10,
+    "Talkshow_train_sample_interval": 10,
+    "bbox_ratio": 1.2,
+    "no_aug": False,
+    "data_strategy": "balance",
+    "total_data_len": 7500000,
+  },
+
+  "train": {
+    "num_gpus": 1,
+    "continue_train": True,
+    "start_over": True,
+    "end_epoch": 20,
+    "train_batch_size": 16,
+    "num_thread": 1,
+    "lr": 1e-5,
+    "min_lr": 1e-6,
+    "save_epoch": 1,
+    "remove_checkpoint": False,
+    "print_iters": 100,
+    "smplx_kps_3d_weight": 100.0,
+    "smplx_kps_2d_weight": 1.0,
+    "smplx_pose_weight": 10.0,
+    "smplx_shape_weight": 1.0,
+    "smplx_orient_weight": 1.0,
+    "hand_root_weight": 1.0,
+    "hand_consist_weight": 1.0,
+  },
+
+  "inference":{
+    "num_gpus": 1,
+    "detection":{
+      "model_type": "yolo",
+      "model_path": "./pretrained_models/yolov8x.pt",
+      "conf": 0.5,
+      "save": False,
+      "verbose": False,
+      "iou_thr": 0.5,
+    },
+    
+  },
+
+  "test": {
+    "test_batch_size": 1
+  },
+
+  "model": {
+    'model_type': 'vit_huge',
+    "pretrained_model_path": './outputs/train_annot_xtp20_20241108_203832/model_dump/snapshot_7.pth.tar',
+    "human_model_path": './human_models/human_model_files',
+    'encoder_pretrained_model_path': './pretrained_models/vitpose_huge.pth',
+    'encoder_config': {
+        'num_classes': 80,
+        'task_tokens_num': 80,
+        'img_size': (256, 192),
+        'patch_size': 16,
+        'embed_dim': 1280,
+        'depth': 32,
+        'num_heads': 16,
+        'ratio': 1,
+        'use_checkpoint': False,
+        'mlp_ratio': 4,
+        'qkv_bias': True,
+        'drop_path_rate': 0.55
+    },
+    'decoder_config': {
+      'feat_dim': 1280,
+      "dim_out": 512,
+      'task_tokens_num': 80,
+    },
+    'input_img_shape': (512, 384),
+    'input_body_shape': (256, 192),
+    'output_hm_shape': (16, 16, 12),
+    'focal': (5000, 5000),
+    'princpt': (192 / 2, 256 / 2),  # virtual principal point position
+    'body_3d_size': 2,
+    'hand_3d_size': 0.3,
+    'face_3d_size': 0.3,
+    'camera_3d_size': 2.5,
+
+  },
+  "log":{
+      'exp_name': None,
+      'output_dir': None,
+      'model_dir': None,
+      'log_dir': None,
+      'result_dir': None,
+  }
+}

smplest_x_h40.pth.tar

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3bd6200c5b2b002842669b733d002e6f5a8cb19ea17daba11f051793bfe384a2
+size 8246417153