NotImplementedError: It is not currently possible to manually set the aspect on 3D axes

Dear @fabro66,

I am encountering the following error. Could you please fix it? Thanks.

python gen_skes.py -v baseball.mp4 -np 1 --animation
Loading YOLOv3 network.....
YOLOv3 network successfully loaded
Loading HRNet model ...
HRNet network successfully loaded
Generating 2D pose ...
100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████| 277/277 [00:28<00:00,  9.77it/s]
Loading GAST-Net ...
/home/orazayy/GAST-Net-3DPoseEstimation-inference/model/local_attention.py:25: UserWarning: This overload of nonzero is deprecated:
        nonzero()
Consider using one of the following signatures instead:
        nonzero(*, bool as_tuple) (Triggered internally at  /opt/conda/conda-bld/pytorch_1603729096996/work/torch/csrc/utils/python_arg_parser.cpp:882.)
  self.e = nn.Parameter(torch.zeros(out_features, len(self.m[0].nonzero()), dtype=torch.float))
GAST-Net successfully loaded
Generating 3D human pose ...
Generating animation ...
Traceback (most recent call last):
  File "gen_skes.py", line 179, in <module>
    generate_skeletons(video=video_path, output_animation=args.animation, num_person=args.num_person)
  File "gen_skes.py", line 153, in generate_skeletons
    render_animation(re_kpts, keypoints_metadata, anim_output, skeleton, 25, 30000, np.array(70., dtype=np.float32),
  File "/home/orazayy/GAST-Net-3DPoseEstimation-inference/tools/vis_h36m.py", line 111, in render_animation
    ax.set_aspect('equal')
  File "/home/orazayy/.conda/envs/wudu/lib/python3.8/site-packages/matplotlib/axes/_base.py", line 1280, in set_aspect
    raise NotImplementedError(
NotImplementedError: It is not currently possible to manually set the aspect on 3D axes

[rirri93]

This is my pip freeze output:

cached-property==1.5.2
cycler==0.10.0
dataclasses==0.8
decorator==4.4.2
ffmpeg==1.4
ffprobe-python==1.0.3
filterpy==1.4.5
future==0.18.2
h5py==3.1.0
imageio==2.9.0
kiwisolver==1.3.1
llvmlite==0.34.0
matplotlib==3.1.0
networkx==2.5
numba==0.51.2
numpy==1.19.4
opencv-python==4.4.0.46
Pillow==8.0.1
pkg-resources==0.0.0
pyparsing==2.4.7
python-dateutil==2.8.1
PyWavelets==1.1.1
PyYAML==5.3.1
scikit-image==0.17.2
scipy==1.5.4
six==1.15.0
tifffile==2020.9.3
torch==1.7.0
torchsummary==1.5.1
torchvision==0.8.1
tqdm==4.52.0
typing-extensions==3.7.4.3
yacs==0.1.8

[yerzhan7orazayev]

@rirri93 many thanks! It now works with your library versions. I really appreciate it.