我正在尝试在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