ImportError: cannot import name 'axis_angle_to_quaternion' from 'pytorch3d.transforms' (/home/yuchen.li/.local/lib/python3.8/site-packages/pytorch3d/transforms/__init__.py)

[yuchenli-sony]

File "test.py", line 15, in
from EDGE import EDGE
File "/group/3d/yuchen/EDGE/EDGE.py", line 15, in
from dataset.dance_dataset import AISTPPDataset
File "/group/3d/yuchen/EDGE/dataset/dance_dataset.py", line 11, in
from pytorch3d.transforms import (RotateAxisAngle, axis_angle_to_quaternion,
ImportError: cannot import name 'axis_angle_to_quaternion' from 'pytorch3d.transforms' (/home/yuchen.li/.local/lib/python3.8/site-packages/pytorch3d/transforms/init.py)

May u share your environment file for pytorch3d library

[yuchenlichuck]

still problem emmmm

[yyxisgreat]

you can use the function from official document of pytorch3d
https://pytorch3d.readthedocs.io/en/latest/_modules/pytorch3d/transforms/rotation_conversions.html#axis_angle_to_quaternion
and add it to your pytorch3d.transforms