nxsEdson / CVD-Physiological-Measurement

Video-based Remote Physiological Measurement via Cross-verified Feature Disentangling. (ECCV2020 oral)
103 stars 23 forks source link

Typo in test.py? #5

Closed Dylan-H-Wang closed 3 years ago

Dylan-H-Wang commented 3 years ago

Hi, In the test() function of test.py, is that a typo when you are writing hr which actually should be bpm?

def test():
    net.eval()
    test_loss = 0;

    for (data, bpm, fps, bvp, idx) in test_loader:

        data = Variable(data);
        hr = Variable(hr.view(-1,1));
        data, hr = data.cuda(), hr.cuda();

        feat_hr, feat_n, output, img_out, feat_hrf1, feat_nf1, hrf1, idx1, feat_hrf2, feat_nf2, hrf2, idx2, ecg, ecg1, ecg2 = net(data, epoch);
        loss = lossfunc_HR(output, hr);

        test_loss += loss.item();
nxsEdson commented 3 years ago

Thank you !! The typo has been corrected.