Update opts_egtea.py

opts_egtea.py

@@ -66,9 +66,12 @@ import os
 
 def parse_opt():
     parser = argparse.ArgumentParser()
-    parser.add_argument('--annotations_path', type=str, default=os.path.join(os.getcwd(), 'data', 'annotations.json'))
+    parser.add_argument('--video_anno', type=str, default=os.path.join(os.getcwd(), 'data', 'annotations.json'))
     parser.add_argument('--video_path', type=str, default=os.path.join(os.getcwd(), 'data', 'videos', 'test_video.mp4'))
-    parser.add_argument('--feature_dir', type=str, default=os.path.join(os.getcwd(), 'data', 'features'))
+    parser.add_argument('--video_feature_all_test', type=str, default=os.path.join(os.getcwd(), 'data', 'features') + os.sep)
+    parser.add_argument('--video_len_file', type=str, default=os.path.join(os.getcwd(), 'data', 'video_len_{}.json'))
+    parser.add_argument('--proposal_label_file', type=str, default=os.path.join(os.getcwd(), 'data', 'proposal_label_{}.h5'))
+    parser.add_argument('--suppress_label_file', type=str, default=os.path.join(os.getcwd(), 'data', 'suppress_label_{}.h5'))
     parser.add_argument('--checkpoint_path', type=str, default=os.path.join(os.getcwd(), 'checkpoints'))
     parser.add_argument('--result_file', type=str, default=os.path.join(os.getcwd(), 'results', 'result_{}.json'))
     parser.add_argument('--frame_result_file', type=str, default=os.path.join(os.getcwd(), 'results', 'frame_result_{}.h5'))
@@ -77,13 +80,13 @@ def parse_opt():
     parser.add_argument('--exp', type=str, default='action_detection')
     parser.add_argument('--mode', type=str, default='test')
     parser.add_argument('--inference_subset', type=str, default='test')
-    parser.add_argument('--num_of_class', type=int, default=106)  # Adjust based on dataset
+    parser.add_argument('--num_of_class', type=int, default=23)  # 22 EGTEA actions + background
     parser.add_argument('--segment_size', type=int, default=16)
     parser.add_argument('--threshold', type=float, default=0.5)
     parser.add_argument('--soft_nms', type=float, default=0.3)
     parser.add_argument('--sup_threshold', type=float, default=0.5)
     parser.add_argument('--anchors', type=str, default='1,2,4,8,16,32')
-    parser.add_argument('--feat_dim', type=int, default=2048)
+    parser.add_argument('--feat_dim', type=int, default=2048)  # rgb (1024) + flow (1024)
     parser.add_argument('--lr', type=float, default=1e-4)
     parser.add_argument('--weight_decay', type=float, default=1e-4)
     parser.add_argument('--lr_step', type=int, default=10)
@@ -93,5 +96,12 @@ def parse_opt():
     parser.add_argument('--gamma', type=float, default=1.0)
     parser.add_argument('--seed', type=int, default=42)
     parser.add_argument('--pptype', type=str, default='nms')
+    parser.add_argument('--data_format', type=str, default='npz_i3d')
+    parser.add_argument('--rgb_only', type=bool, default=False)
+    parser.add_argument('--predefined_fps', type=int, default=30)
+    parser.add_argument('--split', type=str, default='test')
+    parser.add_argument('--setup', type=str, default='default')
+    parser.add_argument('--data_rescale', type=float, default=1.0)
+    parser.add_argument('--pos_threshold', type=float, default=0.5)
     args = parser.parse_args()
     return args