def parse_second_args():
parser = parse_base_args()
parser.add_argument('--data_root', default='/data/yumin/THUMOS', type=str)
parser.add_argument('--model', default='Delta', type=str)
parser.add_argument('--inputs', default='multistream', type=str)
parser.add_argument('--hidden_size', default=4096, type=int)
parser.add_argument('--camera_feature', default='resnet200-fc', type=str)
parser.add_argument('--motion_feature', default='bn_inception', type=str)
parser.add_argument('--enc_steps', default=64, type=int)
# parser.add_argument('--step_size', action='append', help='get list of step size')
parser.add_argument('--dirichlet', action='store_true')
parser.add_argument('--method',
default=None,
type=str,
help=['Mean', 'Sample'])
parser.add_argument('--merge', action='store_true')
parser.add_argument('--var_method',
default=None,
type=str,
help=['covariance', 'diagonal'])
parser.add_argument('--loss_method',
default=None,
type=str,
help=['oad_before', 'state_before'])
return build_data_info(parser.parse_args())
def parse_c3d_args():
parser = parse_base_args()
parser.add_argument('--model', default='C3D', type=str)
parser.add_argument('--epochs', default=100, type=int)
parser.add_argument('--batch_size', default=128, type=int)
parser.add_argument('--lr', default=1e-04, type=float)
parser.add_argument('--weight_decay', default=5e-04, type=float)
return build_data_info(parser.parse_args())
def parse_rnn_args():
parser = parse_base_args()
parser.add_argument('--model', default='RNN', type=str)
parser.add_argument('--features', default='c3d_features', type=str)
parser.add_argument('--epochs', default=100, type=int)
parser.add_argument('--batch_size', default=128, type=int)
parser.add_argument('--lr', default=1e-03, type=float)
return build_data_info(parser.parse_args())
def parse_lstm_args():
parser = parse_base_args()
parser.add_argument('--data_root', default='/data/yumin/THUMOS', type=str)
parser.add_argument('--model', default='LSTM', type=str)
parser.add_argument('--inputs', default='multistream', type=str)
parser.add_argument('--hidden_size', default=4096, type=int)
parser.add_argument('--camera_feature', default='resnet200-fc', type=str)
parser.add_argument('--motion_feature', default='bn_inception', type=str)
parser.add_argument('--enc_steps', default=64, type=int)
return build_data_info(parser.parse_args())
def parse_trn_args():
parser = parse_base_args()
parser.add_argument('--data_root', default='data/THUMOS', type=str)
parser.add_argument('--model', default='TRN', type=str)
parser.add_argument('--inputs', default='multistream', type=str)
parser.add_argument('--hidden_size', default=4096, type=int)
parser.add_argument('--camera_feature', default='resnet200-fc', type=str)
parser.add_argument('--motion_feature', default='bn_inception', type=str)
parser.add_argument('--enc_steps', default=64, type=int)
parser.add_argument('--dec_steps', default=8, type=int)
parser.add_argument('--dropout', default=0.1, type=float)
return build_data_info(parser.parse_args())
def parse_trn_args():
parser = parse_base_args()
parser.add_argument('--data_root', default='data/HDD', type=str)
parser.add_argument('--model', default='TRN', type=str)
parser.add_argument('--inputs', default='multimodal', type=str)
parser.add_argument('--hidden_size', default=2000, type=int)
parser.add_argument('--camera_feature',
default='inceptionresnetv2',
type=str)
parser.add_argument('--enc_steps', default=90, type=int)
parser.add_argument('--dec_steps', default=6, type=int)
parser.add_argument('--dropout', default=0.1, type=float)
return build_data_info(parser.parse_args())
def parse_trn_args():
parser = parse_base_args()
parser.add_argument('--data_root', default='data/CRICKET', type=str)
parser.add_argument('--model', default='TRN', type=str)
parser.add_argument('--inputs', default='motion', type=str)
parser.add_argument('--hidden_size', default=1024, type=int)
# parser.add_argument('--camera_feature', default='resnet200-fc', type=str)
# parser.add_argument('--motion_feature', default='bn-inception', type=str)
# parser.add_argument('--camera_feature', default='hog_64x64', type=str)
parser.add_argument('--camera_feature', default='farneback_flow_grid20_main', type=str)
#parser.add_argument('--motion_feature', default='farneback_flow_grid20', type=str)
# parser.add_argument('--motion_feature', default='c3dFinetuned_feats_17', type=str)
parser.add_argument('--motion_feature', \
default='c3dFinetunedOnHLMainSeq23_mainDataset_feats_17', type=str)
parser.add_argument('--enc_steps', default=64, type=int)
parser.add_argument('--dec_steps', default=8, type=int)
parser.add_argument('--dropout', default=0.1, type=float)
return build_data_info(parser.parse_args())
def parse_demo_args():
parser = parse_base_args()
parser.add_argument('--batch_size', default=1, type=int)
parser.add_argument('--c3d_pth', default='model_zoo/c3d.pth', type=str)
parser.add_argument('--rnn_pth', default='model_zoo/rnn.pth', type=str)
return build_data_info(parser.parse_args())