Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

have you implemented ATI-CNN? #2

Open
nlpCSir opened this issue May 6, 2022 · 5 comments
Open

have you implemented ATI-CNN? #2

nlpCSir opened this issue May 6, 2022 · 5 comments

Comments

@nlpCSir
Copy link

nlpCSir commented May 6, 2022

No description provided.

@wenh06
Copy link
Collaborator

wenh06 commented May 6, 2022

ATI-CNN is essentially a convolutional recurrent neural network (CRNN) with a VGG-like CNN backbone, It can be initiated for example

from copy import deepcopy

from torch_ecg.model_configs import ECG_CRNN_CONFIG
from torch_ecg.models import ECG_CRNN

model_config = deepcopy(ECG_CRNN_CONFIG)
model_config.cnn.name = "vgg16"
# rnn and attention can be changed like above
model = ECG_CRNN(["AF", "PVC"], 12, model_config)

Then the above model is almost an ATI-CNN, except that the attention module proposed in ATI-CNN is not implemented, but you can use the more commonly used se (Squeeze-and-Excitation, the default one), gc (Global Context), nl (Non-Local) instead.

@nlpCSir
Copy link
Author

nlpCSir commented May 10, 2022 via email

@wenh06
Copy link
Collaborator

wenh06 commented May 10, 2022

普通的convolution就是跨channel (lead)的,就把不同channel的信息融合到了一起 (除了所谓的depth-wise convolution, 把groups的数目设置成和in_channels一样。) 另外,ATI-CNN的最后的Attention应该和AttentivePooling是类似的。

@nlpCSir
Copy link
Author

nlpCSir commented Oct 3, 2022

你好作者,关于您工具库的使用,请问我是否可以加您个联系方式,方便交流呢。我在使用的过程中发现了一些问题,如数据库无法加载等

@wenh06
Copy link
Collaborator

wenh06 commented Oct 3, 2022

你好作者,关于您工具库的使用,请问我是否可以加您个联系方式,方便交流呢。我在使用的过程中发现了一些问题,如数据库无法加载等

wenh06@gmail.com, 欢迎提BUG!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants