def extract_h5_paths(current_layer, next_layer):
    """
    Given a layer name, extract the weights, biases, next layer weights and batchnorm params
    """
    path_dict = {}
    layer_idx = current_layer[-1]
    print("changing batchnorm for layer --> ", layer_idx)
    path_dict['wts'] = 'model_weights/' + \
        current_layer + '/' + current_layer + '/' + 'kernel:0'
    path_dict['biases'] = 'model_weights/' + \
        current_layer + '/' + current_layer + '/' + 'bias:0'
    path_dict['next_wts'] = 'model_weights/' + \
        next_layer + '/' + next_layer + '/' + 'kernel:0'
    path_dict['beta'] = 'model_weights/batch_normalization_' + \
        layer_idx + '/batch_normalization_' + layer_idx + '/' + 'beta:0'
    path_dict['gamma'] = 'model_weights/batch_normalization_' + \
        layer_idx + '/batch_normalization_' + layer_idx + '/' + 'gamma:0'
    path_dict['moving_mean'] = 'model_weights/batch_normalization_' + \
        layer_idx + '/batch_normalization_' + layer_idx + '/' + 'moving_mean:0'
    path_dict['moving_variance'] = 'model_weights/batch_normalization_' + \
        layer_idx + '/batch_normalization_' + \
        layer_idx + '/' + 'moving_variance:0'
    return path_dict