update tester

2020-09-10 18:34:52 +08:00
parent 7ea9c6d0d5
commit 72d09aa483
7 changed files with 79 additions and 44 deletions
--- a/configs/synthesizers/TAFG.yml
+++ b/configs/synthesizers/TAFG.yml
@@ -114,6 +114,7 @@ data:
            mean: [ 0.5, 0.5, 0.5 ]
            std: [ 0.5, 0.5, 0.5 ]
  test:
+    which: video_dataset
    dataloader:
      batch_size: 8
      shuffle: False
--- a/configs/synthesizers/TSIT.yml
+++ b/configs/synthesizers/TSIT.yml
@@ -109,6 +109,7 @@ data:
            mean: [ 0.5, 0.5, 0.5 ]
            std: [ 0.5, 0.5, 0.5 ]
  test:
+    which: video_dataset
    dataloader:
      batch_size: 8
      shuffle: False
@@ -116,14 +117,11 @@ data:
      pin_memory: False
      drop_last: False
    dataset:
-      _type: GenerationUnpairedDatasetWithEdge
+      _type: GenerationUnpairedDataset
      root_a: "/data/i2i/VoxCeleb2Anime/testA"
      root_b: "/data/i2i/VoxCeleb2Anime/testB"
-      edges_path: "/data/i2i/VoxCeleb2Anime/edges"
-      landmarks_path: "/data/i2i/VoxCeleb2Anime/landmarks"
-      edge_type: "landmark_hed"
+      with_path: True
      random_pair: False
-      size: [ 128, 128 ]
      pipeline:
        - Load
        - Resize: