3 from demo_superpoint
import SuperPointNet
4 model = SuperPointNet()
5 model.load_state_dict(torch.load(
"superpoint_v1.pth"))
7 example = torch.rand(1, 1, 640, 480)
8 traced_script_module = torch.jit.trace(model, example)
9 traced_script_module.save(
"superpoint_v1.pt")