hukenovs / hagrid

HAnd Gesture Recognition Image Dataset
https://arxiv.org/abs/2206.08219
548 stars 84 forks source link

how to run hand classification #40

Open ardeal opened 1 year ago

ardeal commented 1 year ago

Hi,

In your readme, you mentioned: We provide some pre-trained models as the baseline with the classic backbone architectures and two output heads - for gesture classification and leading hand classification.

I am trying to run resnet18 for hand classification, how to config the yaml file? is the ResNet18 pth model file trained by you for hand classification?

I tried to modify the yaml file, and use ResNet18 for hand classification, but the code report error.

MagicXiaoBai1 commented 5 months ago

下面这是用MobileNetV3_large.pth的过程,和ResNet18 的过程基本一样

1、新建环境、安装requirements.txt中的所有包 2、下载MobileNetV3_large.pth 放在项目目录下(与demo_ff.py同目录) 3、进入configs文件夹并打开MobileNetV3_large.yaml 4、在该文件中找到

model: name: MobileNetV3_large pretrained: False pretrained_backbone: False checkpoint: null 5、将其修改为:

model: name: MobileNetV3_large pretrained: False pretrained_backbone: False checkpoint: MobileNetV3_large.pth 6、执行命令:python demo_ff.py -p configs/MobileNetV3_large.yaml