Skip to content

Commit

Permalink
update readme #1
Browse files Browse the repository at this point in the history
  • Loading branch information
kkoutini authored Jun 7, 2022
1 parent 8428f9e commit ab3440c
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,4 +37,15 @@ embed, time_stamps = get_timestamp_embeddings(audio, model)
print(embed.shape)
embed = get_scene_embeddings(audio, model)
print(embed.shape)
```
```

# Getting the Loggits/Class labels

You can get the logits (before the sigmoid activation) for the 527 classes of audioset:
```pyton
from hear21passt.base import load_model
model = load_model(mode="logits").cuda()
logits = model(wave_signal)
```
The class labels indices can be found [here](https://github.com/qiuqiangkong/audioset_tagging_cnn/blob/master/metadata/class_labels_indices.csv)

0 comments on commit ab3440c

Please sign in to comment.