feat design 功能迁移
This commit is contained in:
14
app/service/design/items/pipelines/segmentation.py
Normal file
14
app/service/design/items/pipelines/segmentation.py
Normal file
@@ -0,0 +1,14 @@
|
||||
from ..builder import PIPELINES
|
||||
from ...utils.design_ensemble import get_seg_result
|
||||
|
||||
|
||||
@PIPELINES.register_module()
|
||||
class Segmentation(object):
|
||||
def __init__(self, device='cpu', show=False, debug=None):
|
||||
self.show = show
|
||||
self.device = device
|
||||
self.debug = debug
|
||||
|
||||
def __call__(self, result):
|
||||
result['seg_result'] = get_seg_result(result["image_id"], result['image'])
|
||||
return result
|
||||
Reference in New Issue
Block a user