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

[Improve] Add exception for PointRend for support CPU-only usage (#1271)

Browse Source 
* [Improve] Add exception for PointRend for support CPU-only usage

* fixed linting
This commit is contained in:
Jerry Jiarui XU 2022-02-09 00:52:56 -05:00 committed by GitHub
parent 66b778c064
commit 82a80880d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmseg/models/decode_heads/point_head.py
View File

@ -4,7 +4,11 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from mmcv.cnn import ConvModule from mmcv.cnn import ConvModule
try:
from mmcv.ops import point_sample from mmcv.ops import point_sample
except ModuleNotFoundError:
point_sample = None
from mmseg.models.builder import HEADS from mmseg.models.builder import HEADS
from mmseg.ops import resize from mmseg.ops import resize
@ -75,6 +79,9 @@ class PointHead(BaseCascadeDecodeHead):
init_cfg=dict( init_cfg=dict(
type='Normal', std=0.01, override=dict(name='fc_seg')), type='Normal', std=0.01, override=dict(name='fc_seg')),
**kwargs) **kwargs)
if point_sample is None:
raise RuntimeError('Please install mmcv-full for '
'point_sample ops')
self.num_fcs = num_fcs self.num_fcs = num_fcs
self.coarse_pred_each_layer = coarse_pred_each_layer self.coarse_pred_each_layer = coarse_pred_each_layer