Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add new Audio metric DNSMOS #2525

Merged
merged 106 commits into from
Jul 17, 2024
Merged
Changes from 1 commit
Commits
Show all changes
106 commits
Select commit Hold shift + click to select a range
e235e35
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 30, 2024
60c7318
+DNSMOS
quancs Apr 30, 2024
213e6ba
update
quancs Apr 30, 2024
fb04f0c
Merge branch 'DNSMOS' of https://github.com/quancs/torchmetrics into …
quancs Apr 30, 2024
284b3aa
fix
quancs Apr 30, 2024
7b01a01
+test
quancs May 1, 2024
0604b7e
fix
quancs May 1, 2024
0da07fe
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 1, 2024
a4749f3
fix
quancs May 1, 2024
572b308
fix
quancs May 1, 2024
fc06a33
fix
quancs May 1, 2024
10c14b2
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 1, 2024
6cff8cc
fix
quancs May 1, 2024
4be8b47
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 1, 2024
f1e1d96
+req
quancs May 1, 2024
6f69898
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 1, 2024
87d9e70
fix
quancs May 2, 2024
4f20381
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 2, 2024
b5ad81e
fix
quancs May 2, 2024
0811527
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 3, 2024
6c52870
fix
quancs May 3, 2024
bdd98ef
fix
quancs May 3, 2024
2862530
fix
quancs May 3, 2024
b6f5200
Merge branch 'DNSMOS' of https://github.com/quancs/torchmetrics into …
quancs May 3, 2024
ba3d1b2
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 3, 2024
32b10f5
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 3, 2024
43b7e5b
fix
quancs May 3, 2024
f1af642
fix
quancs May 3, 2024
b845498
Merge branch 'DNSMOS' of https://github.com/quancs/torchmetrics into …
quancs May 3, 2024
29510ef
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 3, 2024
f96d887
fix
quancs May 3, 2024
1588d51
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 3, 2024
d0a602d
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 3, 2024
99f8810
fix
quancs May 3, 2024
683e5f1
fix
quancs May 3, 2024
8d23eeb
Merge branch 'DNSMOS' of https://github.com/quancs/torchmetrics into …
quancs May 3, 2024
65b8446
fix
quancs May 3, 2024
5048eaf
fix removeprefix
quancs May 3, 2024
b7a8563
Merge branch 'DNSMOS' of https://github.com/quancs/torchmetrics into …
quancs May 3, 2024
d3daa89
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 3, 2024
af48fe2
Apply suggestions from code review
Borda May 6, 2024
d70400f
Merge branch 'master' into DNSMOS
Borda May 6, 2024
808774c
Update CHANGELOG.md
Borda May 6, 2024
21be811
Apply suggestions from code review
Borda May 6, 2024
74cd984
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 6, 2024
91964b2
Merge branch 'master' into DNSMOS
Borda May 7, 2024
9293010
add min & max versions
quancs May 7, 2024
04edab6
fix
quancs May 7, 2024
5eb0dd3
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 7, 2024
baef964
fix
quancs May 8, 2024
e58c4da
fix
quancs May 8, 2024
78c1eef
fix
quancs May 8, 2024
be0cafc
fix
quancs May 8, 2024
2817fc9
fix
quancs May 8, 2024
e6fdb60
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 8, 2024
dba383e
fix
quancs May 11, 2024
e1876cc
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 11, 2024
3a75ea4
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 13, 2024
fe5f54d
fix
quancs May 13, 2024
f24e363
Merge branch 'master' into DNSMOS
Borda May 14, 2024
d69b03c
Merge branch 'master' into DNSMOS
Borda May 19, 2024
7ee68e0
python==3.11, onnxruntime>=1.17.0
quancs May 20, 2024
7796c1a
Merge branch 'DNSMOS' of https://github.com/quancs/torchmetrics into …
quancs May 20, 2024
fa2c2cc
Merge branch 'master' into DNSMOS
Borda May 21, 2024
c6888be
Apply suggestions from code review
Borda May 21, 2024
36e0962
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 21, 2024
bf813a2
with 1.19
Borda May 21, 2024
8575e76
Merge branch 'master' into DNSMOS
Borda May 21, 2024
d77f591
Merge branch 'master' into DNSMOS
Borda May 29, 2024
a248058
Apply suggestions from code review
Borda May 31, 2024
2904d6b
Merge branch 'master' into DNSMOS
Borda May 31, 2024
616f98b
+num_theads
quancs May 31, 2024
0f2b3f8
+P835 paper
quancs May 31, 2024
24a9c46
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 31, 2024
d3b364e
Merge branch 'master' into DNSMOS
Borda Jun 2, 2024
a500116
fix req
quancs Jun 19, 2024
bcdad95
Merge branch 'master' into DNSMOS
quancs Jun 19, 2024
4937e41
fix
quancs Jun 19, 2024
14f0b2a
upgrade pip for resolving onnxruntime installation issue
quancs Jun 19, 2024
1fb40b1
fix onnxruntime==1.18 for python==3.11
quancs Jun 19, 2024
8a9284b
Merge branch 'master' into DNSMOS
quancs Jun 25, 2024
080c76a
Merge branch 'master' into DNSMOS
Borda Jun 25, 2024
f591c90
update
quancs Jun 25, 2024
3c58fb9
Merge branch 'master' into DNSMOS
quancs Jun 25, 2024
bf8e655
Merge branch 'master' into DNSMOS
Borda Jul 6, 2024
7001d9a
Merge branch 'master' into DNSMOS
Borda Jul 9, 2024
fca8d29
Apply suggestions from code review
Borda Jul 9, 2024
6a1e1fb
onnxruntime
quancs Jul 10, 2024
7167b99
fix
quancs Jul 10, 2024
a681818
fix
quancs Jul 10, 2024
5e2591a
use onnxruntime-gpu
quancs Jul 13, 2024
427cc35
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jul 13, 2024
4afc647
Apply suggestions from code review
Borda Jul 13, 2024
7a1d4b7
move to top
quancs Jul 13, 2024
9b1f57b
fix
quancs Jul 13, 2024
9676745
use onnx-gpu for linux & win, onnx for macos
quancs Jul 13, 2024
e627082
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jul 13, 2024
287957c
fix line too long
quancs Jul 14, 2024
9530c70
print available providers
quancs Jul 14, 2024
9e13bf6
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jul 14, 2024
7865b02
+warn if failed to compute DNSMOS on GPU
quancs Jul 14, 2024
7f0ffad
Merge branch 'master' into DNSMOS
quancs Jul 14, 2024
ebd303f
Merge branch 'master' into DNSMOS
Borda Jul 16, 2024
5618f7a
Merge branch 'master' into DNSMOS
Borda Jul 17, 2024
5763494
Merge branch 'master' into DNSMOS
Borda Jul 17, 2024
6037c98
Merge branch 'master' into DNSMOS
mergify[bot] Jul 17, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix
  • Loading branch information
quancs committed May 3, 2024
commit 286253013428be8fe348340c14f82b5515262202
2 changes: 1 addition & 1 deletion src/torchmetrics/functional/audio/dnsmos.py
Original file line number Diff line number Diff line change
@@ -231,7 +231,7 @@ def deep_noise_suppression_mean_opinion_score(preds: Tensor, fs: int, personaliz
input_features = np.array(audio_seg).astype("float32")
p808_input_features = np.array(_audio_melspec(audio=audio_seg[..., :-160])).astype("float32")

if device.type != "cpu":
if device.type != "cpu" and 'CUDAExecutionProvider' in ort.get_all_providers():
input_features = ort.OrtValue.ortvalue_from_numpy(input_features, device.type, device.index)
p808_input_features = ort.OrtValue.ortvalue_from_numpy(p808_input_features, device.type, device.index)