|
|
|
|
@ -32,8 +32,8 @@ def segm_postprocess(box: list, raw_cls_mask, im_h, im_w):
|
|
|
|
|
|
|
|
|
|
class ModelHandler:
|
|
|
|
|
def __init__(self, labels):
|
|
|
|
|
base_dir = os.environ.get("MODEL_PATH",
|
|
|
|
|
"/opt/nuclio/open_model_zoo/public/mask_rcnn_inception_resnet_v2_atrous_coco/FP32")
|
|
|
|
|
base_dir = os.path.abspath(os.environ.get("MODEL_PATH",
|
|
|
|
|
"/opt/nuclio/open_model_zoo/public/mask_rcnn_inception_resnet_v2_atrous_coco/FP32"))
|
|
|
|
|
model_xml = os.path.join(base_dir, "mask_rcnn_inception_resnet_v2_atrous_coco.xml")
|
|
|
|
|
model_bin = os.path.join(base_dir, "mask_rcnn_inception_resnet_v2_atrous_coco.bin")
|
|
|
|
|
self.model = ModelLoader(model_xml, model_bin)
|
|
|
|
|
|