feat: ➕ add tres and update NR benchmark

chaofengc · Feb 23, 2023 · 8d31bf6 · 8d31bf6
1 parent 0afec46
commit 8d31bf6
Show file tree

Hide file tree

Showing 4 changed files with 433 additions and 18 deletions.
diff --git a/options/default_dataset_opt.yml b/options/default_dataset_opt.yml
@@ -66,6 +66,8 @@ spaq:
   type: GeneralNRDataset
   dataroot_target: './datasets/SPAQ/TestImage'
   meta_info_file: './datasets/meta_info/meta_info_SPAQDataset.csv'
+  augment:
+    resize: 448
 
 ava:
   name: AVA
@@ -88,6 +90,7 @@ flive:
   dataroot_target: './datasets/FLIVE_Database/database'
   meta_info_file: './datasets/meta_info/meta_info_FLIVEDataset.csv'
   split_file: './datasets/train_split_info/flive_official.pkl'
+  phase: test
 
 pieapp:
   name: PieAPPDataset