Fix the bug of generating HMDB51 class index file. (#69)

open-mmlab · Jul 29, 2020 · 70b7998 · 70b7998
1 parent 62f55c2
commit 70b7998
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/data/parse_file_list.py b/tools/data/parse_file_list.py
@@ -373,7 +373,7 @@ def generate_class_index_file():
         frame_path = 'data/hmdb51/rawframes'
         annotation_dir = 'data/hmdb51/annotations'
 
-        class_list = os.listdir(frame_path).sort()
+        class_list = sorted(os.listdir(frame_path))
         class_dict = dict()
         with open(class_index_file, 'w') as f:
             content = []
@@ -388,7 +388,7 @@ def generate_class_index_file():
             with open(train_file_template.format(i), 'w') as fout:
                 content = []
                 for class_name in class_dict:
-                    filename = class_name + f'_train_split{i}.txt'
+                    filename = class_name + f'_test_split{i}.txt'
                     filename_path = osp.join(annotation_dir, filename)
                     with open(filename_path, 'r') as fin:
                         for line in fin: