Tensorflow/Keras:一种替代三维卷积的方法

2024-09-22 16:26:10 发布

您现在位置:Python中文网/ 问答频道 /正文

在一些研究工作中,我们应用了一些来自keras的3D Conv层。你知道吗

为了测试我们的移动平台nvidiadrivepx2的训练模型,我们必须将模型转换为TensorRT。但是,当前不支持3D Conv层。有没有办法用2D或1D卷积代替3D Conv层?重新训练模型是可能的。你知道吗


Tags: 模型平台卷积keras办法convtensorrtnvidiadrivepx2