charlie880624/STDC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: not using MMDataParallel for onnx inferencing

Browse Source
This commit is contained in:
chingning.chen 2022-03-23 17:02:03 +08:00
parent 12c840564e
commit b2246495f5
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmseg/models/segmentors/base.py
View File

@ -335,6 +335,9 @@ class ONNXRuntimeSegmentorKN(BaseSegmentor):
def forward_train(self, imgs, img_metas, **kwargs): def forward_train(self, imgs, img_metas, **kwargs):
raise NotImplementedError('This method is not implemented.') raise NotImplementedError('This method is not implemented.')
def forward_test(self, imgs, img_metas, **kwargs):
return super().forward_test(imgs, img_metas[0].data, **kwargs)
@torch.no_grad() @torch.no_grad()
def simple_test(self, img: torch.Tensor, def simple_test(self, img: torch.Tensor,
img_meta: Union[Iterable, None] = None, img_meta: Union[Iterable, None] = None,

2
tools/deploy_test_kneron.py
View File

@ -6,7 +6,6 @@ import warnings
import mmcv import mmcv
import torch import torch
from mmcv.parallel import MMDataParallel
from mmcv.runner import get_dist_info from mmcv.runner import get_dist_info
from mmcv.utils import DictAction from mmcv.utils import DictAction
@ -155,7 +154,6 @@ def main():
else: else:
tmpdir = None tmpdir = None
model = MMDataParallel(model, device_ids=[0])
results = single_gpu_test( results = single_gpu_test(
model, model,
data_loader, data_loader,