DTaoo / VGGish

An implementation of vggish in keras with tf backend
117 stars 38 forks source link

ckpt to h5 #6

Closed sourav22899 closed 5 years ago

sourav22899 commented 5 years ago

@DTaoo The original checkpoint file is provided here. How did you generate a HD5 file without retraining the model ?

beasteers commented 5 years ago

@sourav22899 I have an updated VGGish Keras implementation which includes a conversion script vggish_keras/download_helpers/convert_ckpt.py.

For it to work, you need:

The main difference is that I load weights with by_name=True, so I only create a single weights file (that can be used with or without the top). The weights file also includes the post-processing weights which can optionally be added to the top like in the original implementation.

sourav22899 commented 5 years ago

Thanks for the reply.