[#5] Implement get_*_filename() methods.

Author: just-hjkwon

Diff for: dataset/dataset.py

@@ -59,6 +59,12 @@ def count(self, label):
     def get_labels(self):
         return self.labels
 
+    def get_filename(self, label, index):
+        if self.is_train_mode is True:
+            return self.get_train_filename(label, index)
+        else:
+            return self.get_validation_filename(label, index)
+
     def get_datum(self, label, index):
         if self.is_train_mode is True:
             return self.get_train_datum(label, index)
@@ -82,6 +88,14 @@ def create_pairs_with_json(file_list: list):
 
         return pairs
 
+    @abstractmethod
+    def get_train_filename(self, label: Union[int, str], index: int):
+        pass
+
+    @abstractmethod
+    def get_validation_filename(self, label: Union[int, str], index: int):
+        pass
+
     @abstractmethod
     def train_count(self, label):
         pass

Diff for: dataset/image_dataset.py

@@ -1,6 +1,7 @@
 from .dataset import DataSet
 
 import math
+from typing import Union
 
 import cv2
 import imagesize
@@ -41,6 +42,18 @@ def train_count(self, label):
     def validation_count(self, label):
         return len(self.validation_valid_indices[label])
 
+    def get_train_filename(self, label: Union[int, str], index: int):
+        pair_index = self.train_valid_indices[label][index]
+        image_file_path, json_file_path = self.train_pairs[pair_index]
+
+        return image_file_path, json_file_path
+
+    def get_validation_filename(self, label: Union[int, str], index: int):
+        pair_index = self.validation_valid_indices[label][index]
+        image_file_path, json_file_path = self.validation_pairs[pair_index]
+
+        return image_file_path, json_file_path
+
     def get_train_datum(self, label, index):
         pair_index = self.train_valid_indices[label][index]
 

Diff for: dataset/movie_dataset.py

@@ -2,6 +2,7 @@
 
 import math
 import random
+from typing import Union
 
 import cv2
 import json
@@ -52,6 +53,22 @@ def train_count(self, label):
     def validation_count(self, label):
         return len(self.validation_valid_indices[label].keys())
 
+    def get_train_filename(self, label: Union[int, str], index: int):
+        video_indices = sorted(list(self.train_valid_indices[label].keys()))
+        video_index = video_indices[index]
+
+        movie_file_path, json_file_path = self.train_pairs[video_index]
+
+        return movie_file_path, json_file_path
+
+    def get_validation_filename(self, label: Union[int, str], index: int):
+        video_indices = sorted(list(self.validation_valid_indices[label].keys()))
+        video_index = video_indices[index]
+
+        movie_file_path, json_file_path = self.validation_pairs[video_index]
+
+        return movie_file_path, json_file_path
+
     def get_train_datum(self, label, index):
         video_indices = sorted(list(self.train_valid_indices[label].keys()))
         video_index = video_indices[index]