I3D模型从Tensorflow转移到PyTorch
此仓库包含几个脚本,这些脚本允许从论文的I3D的Tensorflow实现传递权重 Joao Carreira和Andrew Zisserman撰写的PyTorch的。
原始(和官方的!)张量流代码可以在找到。
传输的核心是i3d_tf_to_pt.py脚本
使用python i3d_tf_to_pt.py --rgb启动它,以生成从ImageNet扩展初始化中python i3d_tf_to_pt.py --rgb的rgb检查点权重。
要生成流权重,请使用python i3d_tf_to_pt.py --flow 。
您还可以通过同时使用两个标记python i3d_tf_to_pt.py --rgb --flow在一次运行中生成两者。
请注意,主版本需要PyTorch 0.3,因为它依赖于此最新版本中所包含的Constan
1