本篇博客主要介绍TensorFlow中的迁移学习(transfering learning)。
节约计算资源,在训练好的模型的基础上扩展内容,充分利用原模型的理解力。(迁移前的数据和迁移后的数据相似或相关)
接下来介绍在tensorflow中将VGG16对动物的识别迁移到识别动物的长度。
数据集和VGG已经训练好的权值文件下载地:https://download.youkuaiyun.com/download/github_39611196//11114523/11114523
下面是示例代码:
# encoding:utf-8
from urllib.request import urlretrieve
import os
import numpy as np
import tensorflow as tf
import skimage.io
import skimage.transform
import matplotlib.pyplot as plt
'''
下载训练数据 tiger 和 kittycat