STM-Training:时空记忆网络的训练脚本

时空记忆网络的Trainig脚本 该代码库实现了具有训练代码。 要求 python包 火炬 python-opencv 枕头 yaml ga 雅克 进步 （可选） GPU支持 GPU内存> = 12GB CUDA> = 10.0 数据 有关准备好的数据集的数据组织的更多详细信息，请参见doc 。 释放 我们在代码库中提供了具有不同主干的预训练模型，结果在带有梯度校正的DAVIS17-val上得到了验证。 模型 骨干 数据后端 Ĵ F J＆F 关联 第一人称射击 STM循环 Resnet18 大理 65.3 70.8 68.1 14.8 STM循环 Resnet50 皮尔 70.5 76.3 73.4 9.3 正在运行 将根文件夹附加到python解释器的搜索路径 export PYTHONPATH= ${PYTHONPATH} :./ 要训​​练STM网