File size: 3,063 Bytes
d8dc31c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | config_path = './config/config_ft_pseudogtdataset.py'
use_dposerx = True
dposerx_cfg_path = 'configs.wholebody.subvp.mixed.get_config'
fisheye_camera_path = '/home/cv1/works/SMPLer-X/main/fisheye.calibration_01_12.json'
no_aug = True
use_lora = False
use_smpl = False
num_gpus = 1
exp_name = '/media/cv1/data/output/train_smpler_x_b32'
save_epoch = 1
lr = 1e-05
min_lr = 5e-07
end_epoch = 5
train_batch_size = 8
syncbn = True
bbox_ratio = 1.2
continue_train = True
start_over = True
pretrained_model_path = '/media/cv1/data/output/train_ft_egopw_egowholebody_all_w_3d_positionnet_loss_no_aug_no_bbox_undistort_patch_w_hand_face_20251215_173700/model_dump/snapshot_3.pth.tar'
agora_fix_betas = True
agora_fix_global_orient_transl = True
agora_valid_root_pose = True
dataset_list = [
'Human36M', 'MSCOCO', 'MPII', 'AGORA', 'EHF', 'SynBody', 'GTA_Human2',
'EgoBody_Egocentric', 'EgoBody_Kinect', 'UBody', 'PW3D', 'MuCo', 'PROX',
'PseudoGTDataset', 'Mo2Cap2', 'SceneEgo', 'EgoWholeBody',
'PseudoGTDataset_ext'
]
trainset_3d = []
trainset_2d = []
trainset_humandata = ['PseudoGTDataset']
testset = 'PseudoGTDataset'
use_cache = True
data_strategy = 'concat'
total_data_len = 'auto'
Talkshow_train_sample_interval = 10
fine_tune = None
smplx_loss_weight = 1.0
smplx_pose_weight = 10.0
smplx_kps_3d_weight = 100.0
smplx_kps_2d_weight = 1.0
token_loss_weight = 1.0
net_kps_2d_weight = 1.0
dposer_x_weight = 0.1
dposer_x_weight_min = 0.01
dposer_x_weight_warmup = 0
agora_benchmark = 'agora_model'
model_type = 'smpler_x_h'
encoder_config_file = 'transformer_utils/configs/smpler_x/encoder/body_encoder_huge.py'
encoder_pretrained_model_path = '/home/cv1/works/SMPLer-X/pretrained_models/vitpose_huge.pth'
feat_dim = 1280
upscale = 4
hand_pos_joint_num = 20
face_pos_joint_num = 72
num_task_token = 24
num_noise_sample = 0
train_sample_interval = 10
test_sample_interval = 100
make_same_len = False
input_img_shape = (1024, 1280)
input_body_shape = (256, 192)
output_hm_shape = (16, 16, 12)
input_hand_shape = (256, 256)
output_hand_hm_shape = (16, 16, 16)
output_face_hm_shape = (8, 8, 8)
input_face_shape = (192, 192)
focal = (5000, 5000)
princpt = (640.0, 512.0)
body_3d_size = 2
hand_3d_size = 0.3
face_3d_size = 0.3
camera_3d_size = 0.5
print_iters = 100
lr_mult = 1
test_batch_size = 32
num_thread = 0
vis = True
output_dir = '/media/cv1/data/output/train_smpler_x_b32_20251226_172004'
model_dir = '/media/cv1/data/output/train_smpler_x_b32_20251226_172004/model_dump'
vis_dir = '/media/cv1/data/output/train_smpler_x_b32_20251226_172004/vis'
log_dir = '/media/cv1/data/output/train_smpler_x_b32_20251226_172004/log'
result_dir = '/media/cv1/data/output/train_smpler_x_b32_20251226_172004/result'
code_dir = '/media/cv1/data/output/train_smpler_x_b32_20251226_172004/code'
use_latent_align = True
latent_align_weight = 1.0
cur_dir = '/home/cv1/works/SMPLer-X/main'
root_dir = '/home/cv1/works/SMPLer-X/main/..'
data_dir = '/home/cv1/works/SMPLer-X/main/../dataset'
human_model_path = '/home/cv1/works/SMPLer-X/main/../common/utils/human_model_files'
|