STAM-pytorch:STAM(时空注意模型)的实现,这是一种纯简单的注意模型,可以通过SOTA进行视频分类

时间:2024-04-21 10:15:26
【文件属性】:

文件名称:STAM-pytorch:STAM(时空注意模型)的实现,这是一种纯简单的注意模型,可以通过SOTA进行视频分类

文件大小:61KB

文件格式:ZIP

更新时间:2024-04-21 10:15:26

deep-learning transformers artificial-intelligence attention-mechanism video-classification

STAM-火炬 ,这是又一个纯净且简单的SOTA注意模型,在视频分类中优于所有以前的模型。 这证实了的发现。 注意是我们所需要的。 安装 $ pip install stam-pytorch 用法 import torch from stam_pytorch import STAM model = STAM ( dim = 512 , image_size = 256 , # size of image patch_size = 32 , # patch size num_frames = 5 , # number of image frames, selected out of video space_depth = 12 , # depth of vision transformer space_he


【文件预览】:
STAM-pytorch-main
----setup.py(748B)
----.gitignore(2KB)
----stam.png(56KB)
----LICENSE(1KB)
----stam_pytorch()
--------__init__.py(35B)
--------stam.py(4KB)
----.github()
--------workflows()
----README.md(2KB)

网友评论