DeH4R / README.md
godx7's picture
Update README.md
973778b verified

Model and weights of paper DeH4R: A Decoupled and Hybrid Method for Road Network Graph Extraction.

The pth file contains model architecture and weights, so you should change the way of initializing model like

# model = R2RC(config) do not need this
model = torch.load('xxx.pth')  # make sure you've downloaded the xxx.pth to your local directory
model.to(device)
model.eval()

license: mit