Tensorflow对图像训练进行分类

时间:2017-06-15 07:17:14

标签: image-processing tensorflow

有没有办法在tensorflow中将新类添加到初始v3模型的现有1000个类中?或者是否可以通过向现有类别添加更多图像来微调模型

1 个答案:

答案 0 :(得分:0)

Please see the tutorial How to Retrain Inception's Final Layer for New Categories on TensorFlow.

From the Tutorial:

Modern object recognition models have millions of parameters and can take weeks to fully train. Transfer learning is a technique that shortcuts a lot of this work by taking a fully-trained model for a set of categories like ImageNet, and retrains from the existing weights for new classes. In this example we'll be retraining the final layer from scratch, while leaving all the others untouched. For more information on the approach you can see this paper on Decaf.