diff --git a/ppgan/datasets/vsr_folder_dataset.py b/ppgan/datasets/vsr_folder_dataset.py index 0001fbe37a553642da9552d1ae9a45418af62ad6..b38a5e713e98598fba14d33b641910a0452d5b7d 100644 --- a/ppgan/datasets/vsr_folder_dataset.py +++ b/ppgan/datasets/vsr_folder_dataset.py @@ -56,6 +56,7 @@ class VSRFolderDataset(BaseDataset): def prepare_data_infos(self): sequences = sorted(glob.glob(os.path.join(self.lq_folder, '*'))) + sep = os.path.sep data_infos = [] for sequence in sequences: @@ -67,7 +68,7 @@ class VSRFolderDataset(BaseDataset): data_infos.append( dict(lq_path=self.lq_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, sequence_length=sequence_length)) return data_infos diff --git a/test_tipc/configs/StyleGANv2/train_infer_python.txt b/test_tipc/configs/StyleGANv2/train_infer_python.txt index ec3b238e3af464a1035966c194af518d8d2f601a..9a9b8b42d1d497236ebcdf64978b98d02e18206e 100644 --- a/test_tipc/configs/StyleGANv2/train_infer_python.txt +++ b/test_tipc/configs/StyleGANv2/train_infer_python.txt @@ -20,15 +20,15 @@ distill_train:null null:null null:null ## -===========================eval_params=========================== +===========================eval_params=========================== eval:null null:null ## ===========================infer_params=========================== --output_dir:./output/ 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 -quant_export: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 +quant_export:null fpgm_export:null distill_export:null export1:null