mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Fix mmengine dump of tversky ut in dev-1.x (#2001)

Browse Source
This commit is contained in:
MengzhangLI 2022-08-31 12:01:55 +08:00 committed by GitHub
parent ee7f340dff
commit 8ce7055d1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_models/test_losses/test_tversky_loss.py
View File

@ -36,11 +36,12 @@ def test_tversky_lose():
import os import os
import tempfile import tempfile
import mmcv import mmengine
import numpy as np import numpy as np
tmp_file = tempfile.NamedTemporaryFile() tmp_file = tempfile.NamedTemporaryFile()
mmcv.dump([1.0, 2.0, 3.0], f'{tmp_file.name}.pkl', 'pkl') # from pkl file mmengine.dump([1.0, 2.0, 3.0], f'{tmp_file.name}.pkl',
'pkl') # from pkl file
loss_cfg = dict( loss_cfg = dict(
type='TverskyLoss', type='TverskyLoss',
class_weight=f'{tmp_file.name}.pkl', class_weight=f'{tmp_file.name}.pkl',