|
|
import argparse
|
|
|
|
|
|
def parse_opt():
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
|
|
parser.add_argument('--mode', type=str, default='train')
|
|
|
parser.add_argument('--video_name', type=str, default=None, help='Name of the single video to evaluate')
|
|
|
parser.add_argument('--video_path', type=str, default='', help='Path to the input video file for visualization')
|
|
|
parser.add_argument('--checkpoint_path', type=str, default='./checkpoint')
|
|
|
parser.add_argument('--segment_size', type=int, default=64)
|
|
|
parser.add_argument('--anchors', type=str, default='2,4,6,8,12,16')
|
|
|
parser.add_argument('--seed', default=7, type=int, help='random seed for reproducibility')
|
|
|
|
|
|
|
|
|
parser.add_argument('--num_of_class', type=int, default=23)
|
|
|
parser.add_argument('--data_format', type=str, default="npz_i3d")
|
|
|
parser.add_argument('--data_rescale', default=False, action='store_true')
|
|
|
parser.add_argument('--predefined_fps', default=None, type=float)
|
|
|
parser.add_argument('--rgb_only', default=False, action='store_true')
|
|
|
parser.add_argument('--video_anno', type=str, default="./data/egtea_annotations_split{}.json")
|
|
|
parser.add_argument('--video_feature_all_train', type=str, default="./data/I3D/")
|
|
|
parser.add_argument('--video_feature_all_test', type=str, default="./data/I3D/")
|
|
|
parser.add_argument('--setup', type=str, default="")
|
|
|
parser.add_argument('--exp', type=str, default="01")
|
|
|
parser.add_argument('--split', type=str, default="1")
|
|
|
|
|
|
|
|
|
parser.add_argument('--feat_dim', type=int, default=2048)
|
|
|
parser.add_argument('--hidden_dim', type=int, default=1024)
|
|
|
parser.add_argument('--out_dim', type=int, default=23)
|
|
|
parser.add_argument('--enc_layer', type=int, default=3)
|
|
|
parser.add_argument('--enc_head', type=int, default=8)
|
|
|
parser.add_argument('--dec_layer', type=int, default=5)
|
|
|
parser.add_argument('--dec_head', type=int, default=4)
|
|
|
|
|
|
|
|
|
parser.add_argument('--batch_size', type=int, default=128)
|
|
|
parser.add_argument('--lr', type=float, default=1e-4)
|
|
|
parser.add_argument('--weight_decay', type=float, default=1e-4)
|
|
|
parser.add_argument('--epoch', type=int, default=5)
|
|
|
parser.add_argument('--lr_step', type=int, default=3)
|
|
|
|
|
|
|
|
|
parser.add_argument('--alpha', type=float, default=1)
|
|
|
parser.add_argument('--beta', type=float, default=1)
|
|
|
parser.add_argument('--gamma', type=float, default=0.2)
|
|
|
parser.add_argument('--pptype', type=str, default="net")
|
|
|
parser.add_argument('--pos_threshold', type=float, default=0.5)
|
|
|
parser.add_argument('--sup_threshold', type=float, default=0.1)
|
|
|
parser.add_argument('--threshold', type=float, default=0.1)
|
|
|
parser.add_argument('--inference_subset', type=str, default="test")
|
|
|
parser.add_argument('--soft_nms', type=float, default=0.3)
|
|
|
parser.add_argument('--video_len_file', type=str, default="./output/video_len_{}.json")
|
|
|
parser.add_argument('--proposal_label_file', type=str, default="./output/proposal_label_{}.h5")
|
|
|
parser.add_argument('--suppress_label_file', type=str, default="./output/suppress_label_{}.h5")
|
|
|
parser.add_argument('--suppress_result_file', type=str, default="./output/suppress_result{}.h5")
|
|
|
parser.add_argument('--frame_result_file', type=str, default="./output/frame_result{}.h5")
|
|
|
parser.add_argument('--result_file', type=str, default="./output/result_proposal{}.json")
|
|
|
parser.add_argument('--wterm', type=bool, default=False)
|
|
|
|
|
|
args = parser.parse_args()
|
|
|
return args |
