代码拉取完成，页面将自动刷新
训练一个新任务时，如果从零开始训练时，这将是一个耗时的过程，并且效果可能达不到理想的效果，此时您可以利用PaddleHub提供的预训练模型进行具体任务的Fine-tune。您只需要对自定义数据进行相应的预处理，随后输入预训练模型中，即可得到相应的结果。请参考如下内容设置数据集的结构。
利用PaddleHub迁移分类任务使用自定义数据时，需要切分数据集，将数据集切分为训练集、验证集和测试集。
需要三个文本文件来记录对应的图片路径和标签，此外还需要一个标签文件用于记录标签的名称。
├─data: 数据目录
├─train_list.txt：训练集数据列表
├─test_list.txt：测试集数据列表
├─validate_list.txt：验证集数据列表
├─label_list.txt：标签列表
└─...
训练/验证/测试集的数据列表文件的格式如下
图片1路径 图片1标签
图片2路径 图片2标签
...
label_list.txt的格式如下
分类1名称
分类2名称
...
示例： 以Flower数据集为示例，train_list.txt/test_list.txt/validate_list.txt内容如下示例
roses/8050213579_48e1e7109f.jpg 0
sunflowers/45045003_30bbd0a142_m.jpg 3
daisy/3415180846_d7b5cced14_m.jpg 2
label_list.txt内容如下：
roses
tulips
daisy
sunflowers
dandelion
数据集的准备代码可以参考 flowers.py。hub.datasets.Flowers() 会自动从网络下载数据集并解压到用户目录下$HOME/.paddlehub/dataset目录。具体使用如下：
from paddlehub.datasets import Flowers
flowers = Flowers(transforms)
flowers_validate = Flowers(transforms, mode='val')
transforms: 数据预处理方式。mode: 选择数据模式，可选项有 train, test, val， 默认为train。利用PaddleHub迁移着色任务使用自定义数据时，需要切分数据集，将数据集切分为训练集和测试集。
需要将准备用于着色训练和测试的彩色图像分成训练集数据和测试集数据。
├─data: 数据目录
├─train：训练集数据
|-图片文件夹1
|-图片文件夹2
|-……
|-图片1
|-图片2
|-……
├─test：测试集数据
|-图片文件夹1
|-图片文件夹2
|-……
|-图片1
|-图片2
|-……
└─……
示例：
PaddleHub为用户提供了用于着色的数据集Canvas数据集， 它由1193张莫奈风格和400张梵高风格的图像组成，以Canvas数据集为示例，train文件夹内容如下:
├─train：训练集数据
|-monet
|-图片1
|-图片2
|-……
|-vango
|-图片1
|-图片2
|-……
数据集的准备代码可以参考 canvas.py。hub.datasets.Canvas() 会自动从网络下载数据集并解压到用户目录下$HOME/.paddlehub/dataset目录。具体使用如下：
from paddlehub.datasets import Canvas
color_set = Canvas(transforms, mode='train')
transforms: 数据预处理方式。mode: 选择数据模式，可选项有 train, test, 默认为train。利用PaddleHub进行风格迁移任务使用自定义数据时，需要切分数据集，将数据集切分为训练集和测试集。
需要将准备用于风格迁移的彩色图像分成训练集和测试集数据。
├─data: 数据目录
├─train：训练集数据
|-图片文件夹1
|-图片文件夹2
|-...
|-图片1
|-图片2
|-...
├─test：测试集数据
|-图片文件夹1
|-图片文件夹2
|-...
|-图片1
|-图片2
|-...
|- 21styles
｜-图片1
｜-图片2
└─...
示例：
PaddleHub为用户提供了用于风格迁移的数据集MiniCOCO数据集, 训练集数据和测试集数据来源于COCO2014， 其中训练集有2001张图片，测试集有200张图片。 21styles文件夹下存放着21张不同风格的图片，用户可以根据自己的需求更换不同风格的图片。以MiniCOCO数据集为示例，train文件夹内容如下:
├─train：训练集数据
|-train
|-图片1
|-图片2
|-……
|-test
|-图片1
|-图片2
|-……
|-21styles
|-图片1
|-图片2
|-……
数据集的准备代码可以参考 minicoco.py。hub.datasets.MiniCOCO() 会自动从网络下载数据集并解压到用户目录下$HOME/.paddlehub/dataset目录。具体使用如下：
from paddlehub.datasets import MiniCOCO
ccolor_set = MiniCOCO(transforms, mode='train')
transforms: 数据预处理方式。mode: 选择数据模式，可选项有 train, test, 默认为train。此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。