在Kinetics Pytorch上训练的I3D模型
这个仓库通过Pytorch实现了I3D网络,从张量流转换了预训练的模型权重。
样例代码
您可以将tensorflow模型转换为pytorch
# ./convert
./convert.sh
您可以评估样品
./multi-evaluate.py
与原始模型略有不同。 您可以将原始模型输出与pytorch模型输出在out目录中进行比较
原始模型(imagenet_joint.txt)
Norm of logits: 138.468658
Top classes and probabilities
1.0 41.8137 playing cricket
1.49716e-09 21.494 hurling (sport)
3.84312e-10 20.1341 catching or throwing baseball
1.54923
1