class PolyrnnAdapter(SegBase):
"""Class to interface with polyrnn model."""
def __init__(self, home: str) -> None:
"""Initialize."""
model_dir = os.path.join(home, "experimental", "fast-seg-label")
exp = os.path.join(
model_dir, "polyrnn_scalabel", "Experiments", "tool.json"
)
net = os.path.join(
model_dir, "model_weights", "ggnn_epoch5_step14000.pth"
)
self.tool = Tool(exp, net)
def convert_rect_to_poly(
self, imgs: List[np.ndarray], bboxes: List[List[float]]
) -> List[List[List[float]]]:
"""Convert rectangles to polygons."""
# marshal into segmentation codebase's format
inputs = []
for (img, bbox) in zip(imgs, bboxes):
instance = {"img": img, "bbox": bbox}
component = {"poly": np.array([[-1.0, -1.0]])}
instance = self.tool.preprocess_instance(instance, component)
inputs.append(instance)
# ignore deprecation warnings
with warnings.catch_warnings():
warnings.simplefilter("ignore")
preds: List[List[List[float]]] = self.tool.annotate_batch(inputs)
return preds
class PolyrnnAdapter(SegBase):
"""Class to interface with polyrnn model"""
def __init__(self, home: str) -> None:
model_dir = os.path.join(home, 'experimental', 'fast-seg-label')
exp = os.path.join(model_dir, 'polyrnn_scalabel', 'Experiments',
'tool.json')
net = os.path.join(model_dir, 'model_weights',
'ggnn_epoch5_step14000.pth')
self.tool = Tool(exp, net)
def convert_rect_to_poly(
self, imgs: List[np.ndarray],
bboxes: List[List[float]]) -> List[List[List[float]]]:
# marshal into segmentation codebase's format
inputs = []
for (img, bbox) in zip(imgs, bboxes):
instance = {'img': img, 'bbox': bbox}
component = {'poly': np.array([[-1., -1.]])}
instance = self.tool.preprocess_instance(instance, component)
inputs.append(instance)
# ignore deprecation warnings
with warnings.catch_warnings():
warnings.simplefilter("ignore")
preds: List[List[List[float]]] = self.tool.annotate_batch(inputs)
return preds
def __init__(self, home: str) -> None:
model_dir = os.path.join(home, 'experimental', 'fast-seg-label')
exp = os.path.join(model_dir, 'polyrnn_scalabel', 'Experiments',
'tool.json')
net = os.path.join(model_dir, 'model_weights',
'ggnn_epoch5_step14000.pth')
self.tool = Tool(exp, net)
def __init__(self, home: str) -> None:
"""Initialize."""
model_dir = os.path.join(home, "experimental", "fast-seg-label")
exp = os.path.join(model_dir, "polyrnn_scalabel", "Experiments",
"tool.json")
net = os.path.join(model_dir, "model_weights",
"ggnn_epoch5_step14000.pth")
self.tool = Tool(exp, net)