Error when replicating ViTPose

#5
by AeroDEmi - opened

Hello, I'm trying to create a local machine with ViTPose app, but when making the Pose estimation model I'm getting this error: KeyError: 'ViT is not in the models registry' and KeyError: "TopDown: 'ViT is not in the models registry'". I have tried installing mmcv with mim but I still get the error.

I can load the first model and use it but the pose model is the one I'm struggling to create

Here are my pip installations on my notebook:

!pip install mmcv-full==1.5.0
!pip install mmdet==2.24.1
!pip install  mmpose==0.25.1
!pip install openmim==0.1.5
!pip install timm==0.5.4

And I've tried with mim:

!pip install -U openmim
import mim
mim.uninstall('mmcv-full')
mim.install(['mmcv-full==1.5.0'], is_yes=True)

Solved,
I've just needed mmpose and mmcv_custom from the github repo and add them as custom modules.

AeroDEmi changed discussion status to closed

HI, can you please explain what you did exactly? im stuck at the same point and Im new to cv in general. Much appreciated

Sign up or log in to comment