未验证 提交 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
...@@ -20,15 +20,15 @@ distill_train:null ...@@ -20,15 +20,15 @@ distill_train:null
null:null null:null
null:null null:null
## ##
===========================eval_params=========================== ===========================eval_params===========================
eval:null eval:null
null:null null:null
## ##
===========================infer_params=========================== ===========================infer_params===========================
--output_dir:./output/ --output_dir:./output/
load:null load:null
norm_export:tools/export_model.py -c configs/stylegan_v2_256_ffhq.yaml --inputs_size="1,1,512;1,1" --model_name inference --load norm_export:tools/export_model.py -c configs/stylegan_v2_256_ffhq.yaml --inputs_size="1,1,512;1,1" --model_name inference --load
quant_export:null quant_export:null
fpgm_export:null fpgm_export:null
distill_export:null distill_export:null
export1:null export1:null
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册