未验证 提交 ba4d0651 编写于 作者: W wangna11BD 提交者: GitHub

fix msvsr windows hang (#642)

* fix tipc dataset

* fix msvsr windows hang

* fix msvsr windows hang
上级 bdb0eead
...@@ -56,6 +56,7 @@ class VSRFolderDataset(BaseDataset): ...@@ -56,6 +56,7 @@ class VSRFolderDataset(BaseDataset):
def prepare_data_infos(self): def prepare_data_infos(self):
sequences = sorted(glob.glob(os.path.join(self.lq_folder, '*'))) sequences = sorted(glob.glob(os.path.join(self.lq_folder, '*')))
sep = os.path.sep
data_infos = [] data_infos = []
for sequence in sequences: for sequence in sequences:
...@@ -67,7 +68,7 @@ class VSRFolderDataset(BaseDataset): ...@@ -67,7 +68,7 @@ class VSRFolderDataset(BaseDataset):
data_infos.append( data_infos.append(
dict(lq_path=self.lq_folder, dict(lq_path=self.lq_folder,
gt_path=self.gt_folder, gt_path=self.gt_folder,
key=sequence.replace(f'{self.lq_folder}/', ''), key=sequence.replace(f'{self.lq_folder}' + sep, ''),
num_frames=num_frames, num_frames=num_frames,
sequence_length=sequence_length)) sequence_length=sequence_length))
return data_infos return data_infos
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册