需要参考预训练的3D(非2D图像)分类模型

时间:2019-07-08 14:53:35

标签: tensorflow deep-learning conv-neural-network pre-trained-model

我正在尝试构建3D体积(类似于视频)的二进制对象分类模型,并希望通过从经过良好培训的模型中进行转移学习来节省工作量。我更喜欢使用tensorflow,但愿意使用其他神经网络框架。

我找到了大量代码示例和网站,详细介绍了tensorflow官方网站上的内容 https://www.tensorflow.org/hub/tutorials/image_retraining 但它们都与预训练的2D图像分类模型有关,并且仅为2D模型提供特征提取器层

我发现了3D模型 https://tfhub.dev/s?module-type=video-classification 但是,它们不提供与之配合使用的特征提取器层 因此我只能使用TF Hub上提供的这些3D模型对新视频进行推理,而不能针对我自己的二进制3D对象分类问题进行重新训练

有人可以将我指向带有相关特征提取器层API的预先训练的3D体积模型或视频分类模型,以轻松地进行重新训练并进行迁移学习。

0 个答案:

没有答案