stgcn Namespace Reference

Classes
class	STGCN
class	STGCNBlock

Variables
	model
	loss
	data_loader
	train
	test

Variable Documentation

data_loader

stgcn.data_loader

Initial value:

=  dict(
    type = 'ActionDatasetLoader',
    train_data_folder = '/media/safemotion/HDD5/pjm_test/action_2023_pkl_2',
    test_data_folder = '/media/safemotion/HDD5/pjm_test/action_2023_pkl_2',
    use_normalize = True,
)

Definition at line 27 of file stgcn.py.

loss

stgcn.loss

Initial value:

=  dict(
    CrossEntropyLoss = dict(
        weight = 1.0,
        weights = [1.0],
        pred_keys = ['scores'],
        gt_keys = ['label']
    )
)

Definition at line 18 of file stgcn.py.

model

stgcn.model

Initial value:

=  dict(
    type='STGCNRunner',
    backbone=dict(
        type='STGCN',
        in_channels = 3,
        graph_args = dict(layout = 'coco', strategy='spatial'),
        edge_importance_weighting = True,
        dropout=0.5,
    ),
    head=dict(
        type='STGCNHead',
        in_channels = 256,
        num_class = 17
    ),
    device='cuda:0'
)

Definition at line 1 of file stgcn.py.

test

stgcn.test

Initial value:

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

Definition at line 50 of file stgcn.py.

train

stgcn.train

Initial value:

=  dict(
    num_workers = 8,
    init_lr = 0.01,
    batch_size = 16,
    epochs = 10,
    optimizer = 'SGD',
    optimizer_args = dict(momentum=0.9, nesterov=True, weight_decay=0.0001),
    adjust_lr_epoch = [50, 100, 150, 200],
    adjust_lr_rate = [0.1]*4,    
    val_interval = 1,
    
    pretrained = '/media/safemotion/HDD5/pjm_test/action_train_test/9.pth',
    save_root = '/media/safemotion/HDD5/pjm_test/action_train_test',
 
)

Definition at line 34 of file stgcn.py.