手势动作识别
微调预训练的CNN模型(AlexNet,VGG,ResNet),然后微调LSTM。 该网络应用于手势控制无人机。
训练:
下载直升机编组数据集: : usp
将数据集放在/data文件夹下
运行培训代码并指定数据文件夹的路径
python basic_lstm.py ../data
测试:
使用具有指定型号的网络摄像头运行在线测试代码:
cd testing
python lstm_test.py ../weights/model_best_865.pth.tar
依存关系:
pyTorch-0.3.xx
Opencv的3.3.1
PIL-5.0.0
Numpy-1.13.1
2021-10-06 15:47:51
35.82MB
Python
1