Audio-indexing off by one step

This commit is contained in:
Prajwal 2020-08-31 16:12:32 +05:30 committed by GitHub
parent 1e903f5658
commit d27e32bad0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -55,7 +55,7 @@ class Dataset(object):
def crop_audio_window(self, spec, start_frame):
# num_frames = (T x hop_size * fps) / sample_rate
start_frame_num = self.get_frame_id(start_frame) + 1 # 0-indexing ---> 1-indexing
start_frame_num = self.get_frame_id(start_frame)
start_idx = int(80. * (start_frame_num / float(hparams.fps)))
end_idx = start_idx + syncnet_mel_step_size
@ -278,4 +278,4 @@ if __name__ == "__main__":
train(device, model, train_data_loader, test_data_loader, optimizer,
checkpoint_dir=checkpoint_dir,
checkpoint_interval=hparams.syncnet_checkpoint_interval,
nepochs=hparams.nepochs)
nepochs=hparams.nepochs)