Closed xiao-mb closed 3 years ago
Hi @xiao-mb 后续我们会添加全连接层。
目前,用户可以暂时使用 Conv2D 代替 全连接层。因为 ESP-DL 目前主要支持特征图的形状为(H,W,C),所以我假设你的全连接层的输入来自 Conv2D 的输出。有以下两种情况:
方法一:如果 Conv2D 的输出形状为(1,1,C),那么你可以用以下方法实现全连接层:
方法二:如果 Conv2D 的输出形状为(H,W,C),其中 H!=1,W!=1,情况相对复杂一些:
Tensor fcn_input(preview_output, true);
fcn_input.set_shape({1,1,H*W*C});// 相当于 flatten
fcn_output = nn::conv2d(...);
注意:
如果全连接层的输入来自GlobalAveragePool2D的输出,要如何操作?
同 “输入来自 Conv2D”
库里没看到全连接层的接口,请问如何实现?