上传者: 38576045
|
上传时间: 2022-03-22 06:07:27
|
文件大小: 33.85MB
|
文件类型: -
用卷积滤波器matlab代码TCN的眼动分类
用于三级眼动分类问题的时间卷积网络(3EMCP)
该存储库提供对代码库,模型和评估结果的访问,该论文在论文《带有时间卷积网络的眼动分类》中进行了描述(将很快提供链接)。
请注意,此处共享的大多数代码最初是由实施的。
我们仅添加了对TCN的支持,将其升级到Python
3,实现了一些新工具,并在Python中提供了一个新的功能提取器(原始功能提取器在MATLAB中)。
配置
为了训练新的基于TCN的模型或评估先前训练的模型,您需要首先下载一些包含所有必要数据的大型压缩文件,然后根据以下说明进行提取:
下载,其中包含GazeCom预先计算的功能(具有额外的比例和功能),并将其提取到data/inputs/
下载具有所有受过训练的模型的压缩文件,并将其解压缩到存储库根文件夹中
下载,将文件包含经过训练的模型的生成的输出进行评估,然后将其提取到存储库根文件夹中
已知依赖
Python
3.6+
TensorFlow
2.0+
脾气暴躁的
训练
要训练新的TCN模型,应运行train_tcn.py脚本,但首先需要设置训练参数。
这是在代码中完成的