文件名称：TimeSformer-pytorch：TimeSformer的实现，这是一种基于注意力的纯视频分类解决方案
文件大小：152KB
文件格式：ZIP
更新时间：2024-03-24 04:33:19
deep-learning transformers artificial-intelligence attention-mechanism video-classification
TimeSformer-Pytorch 实现，是一种基于关注点的纯净，简单的解决方案，可以在视频分类上达到SOTA。 该存储库将仅存储性能最佳的变体“时空分散注意力”，无非就是沿空间之前的时间轴的注意力。 安装 $ pip install timesformer-pytorch 用法 import torch from timesformer_pytorch import TimeSformer model = TimeSformer ( dim = 512 , image_size = 224 , patch_size = 16 , num_frames = 8 , num_classes = 10 , depth = 12 , heads = 8 , dim_head = 64 , attn_dropout =
【文件预览】：
TimeSformer-pytorch-main
----timesformer_pytorch()
--------timesformer_pytorch.py(5KB)
--------__init__.py(64B)
----.github()
--------workflows()
----LICENSE(1KB)
----setup.py(763B)
----.gitignore(2KB)
----diagram.png(153KB)
----README.md(1KB)