dmsm / learningpatches Goto Github PK

Hello,

As the title suggests, I'm trying to generate patches using the Sphere templates as mentioned in the paper. I can see in the scripts/run.py file that by default the category specific templates are used. Replacing the template paths to use the sphere24/54 folder, raises Unable to load checkpoint error.

For example, changing the following line in main method of scripts/run.py

edge_data = [line.strip().split(' ') for line in open(
        os.path.join('templates', args.category, 'edges.txt'), 'r')]

to

edge_data = [line.strip().split(' ') for line in open(
        os.path.join('templates', 'sphere24', 'edges.txt'), 'r')]

and similarly changing paths of variables junction_order, topology, adjacencies, vertex_idx, and processed_vertices

raises

Unable to load checkpoint
torch.Size([3, 128, 128])

Please let me know if any additional information is required. Any guidance about generating patches with sphere templates will be appreciated . Thank you!