From ed3c3599f4b2fe978442181098b860c7132992ec Mon Sep 17 00:00:00 2001 From: huangjun12 <12272008@bjtu.edu.cn> Date: Wed, 29 Apr 2020 03:12:31 +0000 Subject: [PATCH] fix bmn bug, data type missmatch in windows --- examples/bmn/reader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/bmn/reader.py b/examples/bmn/reader.py index 58fd21f..d8f125e 100644 --- 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 -- GitLab