Skip to content

Commit

Permalink
minor update
Browse files Browse the repository at this point in the history
  • Loading branch information
zhaodongsun authored Aug 9, 2022
1 parent 9f8febf commit a6153c5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ def my_config():
e = 29 # the model checkpoint at epoch e
train_exp_num = 1 # the training experiment number
train_exp_dir = './results/%d'%train_exp_num # training experiment directory
time_interval = 30 # evaluate 30s clips
time_interval = 30 # get rppg for 30s video clips, too long clips might cause out of memory

ex.observers.append(FileStorageObserver(train_exp_dir))

Expand Down Expand Up @@ -80,4 +80,4 @@ def dl_model(imgs_clip):
# bvppeak_list = np.array(bvppeak_list)
# results = {'rppg_list': rppg_list, 'bvp_list': bvp_list, 'bvppeak_list':bvppeak_list}
results = {'rppg_list': rppg_list, 'bvp_list': bvp_list}
np.save(pred_exp_dir+'/'+h5_path.split('/')[-1][:-3], results)
np.save(pred_exp_dir+'/'+h5_path.split('/')[-1][:-3], results)

0 comments on commit a6153c5

Please sign in to comment.