未验证 提交 581b4944 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #65 from huangjun12/fix-bmn-reader

fix bmn bug, data type missmatch in windows
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册