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 typo in warning on binary classification (#2382)

Browse Source
This commit is contained in:
nulam 2022-12-06 04:29:11 +01:00 committed by GitHub
parent f3db8d779d
commit 3bac6ef5a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmseg/models/decode_heads/decode_head.py
View File

@ -120,7 +120,7 @@ class BaseDecodeHead(BaseModule, metaclass=ABCMeta):
warnings.warn('For binary segmentation, we suggest using' warnings.warn('For binary segmentation, we suggest using'
'`out_channels = 1` to define the output' '`out_channels = 1` to define the output'
'channels of segmentor, and use `threshold`' 'channels of segmentor, and use `threshold`'
'to convert seg_logist into a prediction' 'to convert `seg_logits` into a prediction'
'applying a threshold') 'applying a threshold')
out_channels = num_classes out_channels = num_classes