从概念上讲,如何将三个具有不同核大小的全连接层合并为一个新的全连接层?

我正在尝试使用Tensorflow编写这里给出的gazecapture模型的代码:

http://gazecapture.csail.mit.edu/cvpr2016_gazecapture.pdf

在模型的末尾,眼睛数据、面部数据和面部网格数据已经被简化为三个不同的全连接层,分别为1×128、1×64和1×128大小。权重似乎没有共享,但它们以某种方式结合成一个最终的全连接层。我缺少将这些层连接起来形成最终全连接层的链接。

这是如何实现的?其他地方有这方面的例子吗?

如果有帮助的话,原始项目的github位于这里:

https://github.com/CSAILVision/GazeCapture

谢谢你!如果这是一个初级错误,我提前道歉。


回答:

很好,所以眼睛、面部和面部网格数据已经各自简化为各自的全连接层,这使得事情变得简单。我们可以使用Tensorflow中的内置函数进行拼接,如下所示

big_fcl = tf.concat([eye_fcl, face_fcl, face_grid_fcl], 1)

(‘fcl’代表’全连接层’)

现在big_fcl的大小为128+64+128。

完成!

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注