时空记忆网络的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网
2023-02-15 21:47:28
36.55MB
Python
1