equi-pose icon indicating copy to clipboard operation
equi-pose copied to clipboard

Error in vgtk.functional.rotation.py

Open GitHubOKay opened this issue 2 years ago • 0 comments

thanks for your excellent work. I get a problem when I test the model you provided. Here is the error: adj_idx: [array([[4, 1, 5]]), array([[0, 2]]), array([[1, 3]]), array([[2, 4, 8]]), array([[0, 3, 9]]), array([[ 0, 10]]), array([[11, 12]]), array([[12, 13]]), array([[ 3, 14]]), array([[ 4, 10, 14]]), array([[ 9, 5, 15]]), array([[ 6, 16]]), array([[ 6, 7, 17]]), array([[ 7, 18]]), array([[ 9, 8, 19]]), array([[10, 19, 16]]), array([[11, 15, 17]]), array([[12, 16, 18]]), array([[13, 17, 19]]), array([[14, 15, 18]])] Traceback (most recent call last): File "main.py", line 10, in from datasets.dataset_parser import DatasetParser File "/home/sh/jun/equi-pose/datasets/dataset_parser.py", line 18, in from datasets.modelnet40_complete import ModelNet40Complete # complete File "/home/sh/jun/equi-pose/datasets/modelnet40_complete.py", line 9, in import vgtk.pc as pctk File "/home/sh/jun/equi-pose/models/vgtk/init.py", line 5, in from . import so3conv File "/home/sh/jun/equi-pose/models/vgtk/so3conv/init.py", line 2, in from .functional import * File "/home/sh/jun/equi-pose/models/vgtk/so3conv/functional.py", line 387, in Rs, R_idx, canonical_relative = fr.icosahedron_so3_trimesh(ANCHOR_PATH, GAMMA_SIZE) File "/home/sh/jun/equi-pose/models/vgtk/functional/rotation.py", line 260, in icosahedron_so3_trimesh R_adj = get_adjmatrix_trimesh(mesh, gsize) File "/home/sh/jun/equi-pose/models/vgtk/functional/rotation.py", line 128, in get_adjmatrix_trimesh face_adj = np.vstack(adj_idx).astype(np.int32) File "<array_function internals>", line 6, in vstack File "/home/sh/anaconda3/envs/jun_pvn3d/lib/python3.6/site-packages/numpy/core/shape_base.py", line 283, in vstack return _nx.concatenate(arrs, 0) File "<array_function internals>", line 6, in concatenate ValueError: all the input array dimensions for the concatenation axis must match exactly, but along dimension 1, the array at index 0 has size 3 and the array at index 1 has size 2

The error may be due to 'vstack' receiving arrays of different dimensions. Could you please give me some advice to solve this problem? Thanks!

GitHubOKay avatar Nov 10 '22 04:11 GitHubOKay