feat(save): support save eval_index frames results only to .h5 file.

Author: Kin-Zhang

conf/save.yaml

@@ -1,7 +1,7 @@
 dataset_path: /home/kin/data/av2/h5py/demo/val
 checkpoint: /home/kin/model_zoo/seflow_best.ckpt
 res_name: # if None will directly be the `model_name.ckpt` in checkpoint path
-
+val_index_only: False # [True, False]
 
 # no need to change
 num_frames: 2

save.py

@@ -54,7 +54,7 @@ def main(cfg):
     # NOTE(Qingwen): search & check in pl_model.py : def test_step(self, batch, res_dict)
     trainer.test(model = mymodel, \
                  dataloaders = DataLoader(\
-                     HDF5Dataset(cfg.dataset_path, n_frames=cfg.num_frames), \
+                     HDF5Dataset(cfg.dataset_path, n_frames=cfg.num_frames, eval=cfg.val_index_only), \
                     batch_size=1, shuffle=False))
 
 if __name__ == "__main__":