Load id2label from json

run_opencv.py

@@ -1,3 +1,4 @@
+import json
 from typing import List, Optional, Tuple
 
 import cv2
@@ -32,6 +33,8 @@ class ArgParser(Tap):
 
     top_k: Optional[int] = 5
 
+    id2label: Optional[str] = "labels/kinetics_400.json"
+
 
 class ActivityModel:
     def __init__(self, args: ArgParser):
@@ -41,6 +44,13 @@ class ActivityModel:
         self.frames_per_video = self.get_frames_per_video(args.model_name)
         print(f"Frames per video: {self.frames_per_video}")
 
+        self.load_json()
+
+    def load_json(self):
+        if args.id2label is not None:
+            with open(args.id2label, encoding="utf-8") as f:
+                self.model.config.id2label = json.load(f)
+
     def load_model(
         self, model_name: str
     ) -> Tuple[VideoMAEFeatureExtractor, TimesformerForVideoClassification]: