fix bmn bug, data type missmatch in windows

ed3c3599 · huangjun12 · 61e218d0 · ed3c3599
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

examples/bmn/reader.py examples/bmn/reader.py +1 -1

未找到文件。
--- a/examples/bmn/reader.py
+++ b/examples/bmn/reader.py
@@ -48,7 +48,7 @@ class BmnDataset(Dataset):

    def __getitem__(self, index):
        video_name = self.video_list[index]
-        video_idx = self.video_list.index(video_name)
+        video_idx = np.array(self.video_list.index(video_name)).astype('int64')
        video_feat = self.load_file(video_name)
        if self.mode == 'infer':
            return video_feat, video_idx