纯视频分类解决方案【源码】

TimeSformer-pytorch：TimeSformer的实现，这是一种基于注意力的纯视频分类解决方案-源码
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.zip (151.55 KB, 下载次数: 10)

2021-5-13 14:59 上传

点击文件名下载附件