Exemplo n.º 1
0
 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)
Exemplo n.º 2
0
 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)