yochaiye / scene_agnostic_dereverberation

PyTorch implementation of the 2021 INTERSPEECH paper "Scene-Agnostic Multi-Microphone Speech Dereverberation"
8 stars 5 forks source link

Pretrained model and unavailability of data to replicate results #1

Open ajinkyakulkarni14 opened 2 years ago

ajinkyakulkarni14 commented 2 years ago

Hello, THank you for release of the code. Will it be possible to release the pretrained model?

Also, as the WSJ0 dataset is not free available, it is difficult to run the code. Will it be possible to state the shape of input tensor and output tensor?

Thank you.

yochaiye commented 2 years ago

@ajinkyakulkarni14 Hi, Sorry for only getting back to you now!

I will be able to upload the trained model in the next few days. Any preference of microphones number? Noisy/Noiseless dataset?

As for the second question - unfortunately you must have the WSJ0 in order to create the dataset. Anyway, the input shape is BATCH_SIZE x MICS_NUM x 256 x 256 x 1, and the output shape is BATCH_SIZE x 256 x 256 x 1

ajinkyakulkarni14 commented 2 years ago

Thank you so much for your reply. I am working with three microphones with Noisy dataset in Far field scenario.

Also, can you also explain input shape and output shape, is it batch_size x mics_num x frame_len x feat_dim x 1 or batch_size x mics_num x feat_dim x frame_len x 1??

yochaiye commented 2 years ago

From what I see, the format at the input is batch_size x mics_num x time x frequency. I will initiate a model training a model and then upload it. By the way, if 4 microphones is ok for you, I have an off-the-shelf model to upload. Let me know

yochaiye commented 2 years ago

@ajinkyakulkarni14 I trained a model, but it is too large to upload. Have you got a google drive account?

ajinkyakulkarni14 commented 2 years ago

@yochaiye Yes, I have google drive. Also, Can you share your email id? My email id is ajinkyakulkarni14@gmail.com

yochaiye commented 2 years ago

@ajinkyakulkarni14 I have sent you a link, you can check your inbox.

ajinkyakulkarni14 commented 2 years ago

@yochaiye I have received the model thanks for the help.