Safemotion Lib
Loading...
Searching...
No Matches
Variables
stgcn_mc2 copy Namespace Reference

Variables

 model
 
 loss
 
 data_loader
 
 train
 
 test
 

Variable Documentation

◆ data_loader

stgcn_mc2 copy.data_loader
Initial value:
1= dict(
2 type = 'ActionDatasetLoader',
3 train_data_folder = '/media/safemotion/HDD5/pjm_test/action_mc_train/1st/train',
4 test_data_folder = '/media/safemotion/HDD5/pjm_test/action_mc_train/1st/val',
5 use_normalize = True,
6)

Definition at line 28 of file stgcn_mc2 copy.py.

◆ loss

stgcn_mc2 copy.loss
Initial value:
1= dict(
2 CrossEntropyLoss = dict(
3 weight = 1.0,
4 weights = [1.0, 1.0],
5 pred_keys = ['pose', 'action'],
6 gt_keys = ['label_pose', 'label_action'],
7 data_num = None
8 )
9)

Definition at line 18 of file stgcn_mc2 copy.py.

◆ model

stgcn_mc2 copy.model
Initial value:
1= dict(
2 type='MCSTGCNRunner',
3 backbone=dict(
4 type='STGCN',
5 in_channels = 3,
6 graph_args = dict(layout = 'coco', strategy='spatial'),
7 edge_importance_weighting = True,
8 dropout=0.5,
9 ),
10 head=dict(
11 type='STGCNMCHead',
12 mc_params = dict(pose = dict(in_channels=256, num_class=4),
13 action = dict(in_channels=256, num_class=9))
14 ),
15 device='cuda:0'
16)

Definition at line 1 of file stgcn_mc2 copy.py.

◆ test

stgcn_mc2 copy.test
Initial value:
1= dict(
2 model_path = '',
3 save_root = '',
4)

Definition at line 51 of file stgcn_mc2 copy.py.

◆ train

stgcn_mc2 copy.train
Initial value:
1= dict(
2 num_workers = 8,
3 init_lr = 0.1,
4 batch_size = 32,
5 epochs = 3000,
6 optimizer = 'SGD',
7 optimizer_args = dict(momentum=0.9, nesterov=True, weight_decay=0.0001),
8 adjust_lr_epoch = [500, 1000, 1500, 2000],
9 adjust_lr_rate = [0.1]*4,
10 val_interval = 1,
11
12 pretrained = None,#'/media/safemotion/HDD5/pjm_test/action_train_test2/9.pth',
13 save_root = '/media/safemotion/HDD5/pjm_test/action_train_mc_test2',
14
15)

Definition at line 35 of file stgcn_mc2 copy.py.