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
2021-06-29 11:46:53 319.6MB pytorch weight kinetics 3d-convolutional-network
1