Custom flattenLayer compatible with imageInputLayer

Question

Amelia Moseley 2020-9-19

2
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/596500-custom-flattenlayer-compatible-with-imageinputlayer

回答： Dan 2022-9-8

I have been reading through some papers as part of research for my BSc major project, and came across this paper https://www.ijeat.org/wp-content/uploads/papers/v8i6/F8602088619.pdf outlining a hybrid LSTM-CNN architecture. While not the main subject of my project, I am curious about implementing the basic diagram that they have listed.

The architecture is simple enough: image input, batch normalization, lstm, convolution, max pool, fully connected, output.

Obviously a flatten layer is needed between batch norm and lstm, however the flatten layer provided in matlab is not compatible with image input layers (both 2D and 3D).

Reading the Flatten.m source file, the comments list the basic details of image dimensions, however the FlattenLayer.m (class) and flattenLayer.m (function), only list sequence data dimensions in their comments.

This makes me wonder if it is at all possible to make a custom flatten layer that is compatible with image input.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Madhav Thakker 2020-9-23

3
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/596500-custom-flattenlayer-compatible-with-imageinputlayer#answer_498871

Hi Amelia, I understand that you want to use flatten layer with Image input layer. As of now, faltten layer supports sequence input only. This exhancement is being taken into consideration and the concerned parties may be investigating further.

You can always define your own custom deep learning layers if required. https://www.mathworks.com/help/deeplearning/ug/define-custom-deep-learning-layers.html

Hope this helps.

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Shino Asada 2021-3-29

I have a simple CNN network that uses FlattenLayer, which prevents me from porting the network to MATLAB to run. If MATLAB implements ReshapeLayer, I believe this will reduce a lot of trouble.

请先登录，再进行评论。