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())