posec3d_pose_fl Namespace Reference

Variables
	model
	loss
	metric_args
list	collect_keys = ['pose_heatmap', 'label']
	data_loader
int	ep_mul = 20
	train
	test

Variable Documentation

collect_keys

list posec3d_pose_fl.collect_keys = ['pose_heatmap', 'label']

Definition at line 70 of file posec3d_pose_fl.py.

data_loader

posec3d_pose_fl.data_loader

Initial value:

=  dict(
    type = 'ActionDatasetLoader_v22',
    data_folder = '/media/safemotion/HDD5/pjm_test/action_train_dataset_2023/action_v22_1st_split',
    category_info = dict(pose = 8,
                         hand = 2,
                         foot = 2),
    clip_len = 8,
)

Definition at line 72 of file posec3d_pose_fl.py.

ep_mul

int posec3d_pose_fl.ep_mul = 20

Definition at line 80 of file posec3d_pose_fl.py.

loss

posec3d_pose_fl.loss

Initial value:

=  dict(
    MutiTaskSigmoidFocalLoss = dict(
        weight = 1.0,
        task_key = 'category',
        pred_keys = ['pose', 'hand', 'foot'],
        gt_keys = ['label', 'label', 'label'],
        target_tasks = ['pose', 'hand', 'foot'],
        weights = [1.0, 1.0, 1.0],
    )
)

Definition at line 54 of file posec3d_pose_fl.py.

metric_args

posec3d_pose_fl.metric_args

Initial value:

=  dict(pred_key=['pred_pose', 'pred_hand', 'pred_foot'], 
                   gt_key=['label', 'label', 'label'], 
                   target_tasks=['pose', 'hand', 'foot'],
                   task_key='category')

Definition at line 64 of file posec3d_pose_fl.py.

model

posec3d_pose_fl.model

Definition at line 1 of file posec3d_pose_fl.py.

test

posec3d_pose_fl.test

Initial value:

=  dict(
    model_path = '',
    save_root = '',
)

Definition at line 105 of file posec3d_pose_fl.py.

train

posec3d_pose_fl.train

Initial value:

=  dict(
    num_workers = 8,
    init_lr = 0.2,
    batch_size = 32,
    epochs = 150*ep_mul,
    optimizer = 'SGD',
    optimizer_args = dict(momentum=0.9, nesterov=True, weight_decay=0.0001),
    scheduler = 'CosineAnnealingLR',
    scheduler_args = dict(T_max=30*ep_mul, eta_min=0),
    adjust_lr_epoch = [30*ep_mul, 90*ep_mul, 150*ep_mul],
    adjust_lr_rate = [0.5]*3,
    val_interval = 1,
    
 
    update_loss_weight = False,
    update_loss_weight_interval = 20,
    base_weight = 0.5,
 
 
    pretrained = None,#'/media/safemotion/HDD5/pjm_test/action_train_test/9.pth',
    save_root = '/media/safemotion/HDD5/pjm_test/action_train_result/pose_1st_fl',
 
)

Definition at line 81 of file posec3d_pose_fl.py.