search

sotiraslab / AgileFormer

This the repo for the paper tiltled "AgileFormer: Spatially Agile Transformer UNet for Medical Image Segmentation"
https://arxiv.org/abs/2404.00122
Apache License 2.0
63 stars 9 forks source link

MLP layer #13

Closed zhangxuan-thecastle closed 3 months ago

zhangxuan-thecastle commented 4 months ago

Dear author, may I ask if there is an MLP layer in your code? I did not find the MLP layer

peijie-chiu commented 4 months ago

@zhangxuan-thecastle Thank you for being interested in our work. The MLP layer actually implemented as convolutional layers. You can find it in the "networks/block_2d.py": https://github.com/sotiraslab/AgileFormer/blob/main/networks/blocks_2d.py. Let me know if this answers your questions.