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)