如何使输入通道大于3的图像堆栈的张量流初始模型工作?

时间:2018-04-28 03:41:22

标签: python tensorflow computer-vision

我正在尝试在TensorFlow Inception中重新实现https://github.com/jeffreyhuang1/two-stream-action-recognition(PyTorch ResNet)。在构建使用20通道图像堆栈的模型时,代码是直截了当的!

self.model = resnet101(pretrained = True,channel = 20).cuda()

虽然在TensorFlow中没有规定修改输入通道的数量。我可以想到的解决方法是导入在imagenet数据(RGB,3通道)上预先训练的初始模型,并将所需的变量添加到模型图中以容纳20通道输入。我不知道这是否可以在TensorFlow中完成。

类似的问题:Input channels equal to 6 on tensorflow

和一个教程,其内容不是很丰富:https://github.com/minhnhat93/tf_object_detection_multi_channels

0 个答案:

没有答案