[视觉模型]迁移学习之五个步骤

一.概要

1.组织数据

2.构建数据流水线

3.数据增强

4.定义模型

5.训练和测试

二.详情

1.组织数据

a.下载好带标注的数据

b.将数据劈成三部分(80%用在训练,10%用在验证,10%用在测试)

2.构建数据流水线

a.定义好读取数据的流水线(包括前处理/组batch)

3.数据增强

a.包括旋转/缩放等

4.定义模型

a.基于已有的模型,删除和任务相关的层,附加上新的层.

b.维持原有层的权重

c.选择合适的优化方法

d.选择合适的度量方法

5.训练和测试

a.迭代训练直到验收的精度和性能符合预取

b.保存模型供应用使用