ModuleNotFoundError: No module named 'imp_model

fabro66 / GAST-Net-3DPoseEstimation

A Graph Attention Spatio-temporal Convolutional Networks for 3D Human Pose Estimation in Video (GAST-Net)

MIT License

313 stars 70 forks source link

ModuleNotFoundError: No module named 'imp_model #12

Closed yerzhan7orazayev closed 3 years ago

yerzhan7orazayev commented 3 years ago

Dear @fabro66,

When running gen_skes.py I am encountering the following error. Could you please fix it? Thanks a lot.

python gen_skes.py -v baseball.mp4 -np 1 --animation
Traceback (most recent call last):
  File "gen_skes.py", line 14, in <module>
    from imp_model.gast_net import SpatioTemporalModelOptimized1f
ModuleNotFoundError: No module named 'imp_model

fabro66 commented 3 years ago

Hi~ I have fixed the bug.

Replace

from model.gast_net import SpatioTemporalModel
from imp_model.gast_net import SpatioTemporalModelOptimized1f

with

from model.gast_net import SpatioTemporalModel, SpatioTemporalModelOptimized1f
# from imp_model.gast_net import SpatioTemporalModelOptimized1f