search

PaddlePaddle / PaddleHub

Awesome pre-trained models toolkit based on PaddlePaddle. (400+ models including Image, Text, Audio, Video and Cross-Modal with Easy Inference & Serving)【安全加固，暂停交互，请耐心等待】
https://www.paddlepaddle.org.cn/hub
Apache License 2.0
12.75k stars 2.07k forks source link

如何在hub2.x的finetune中加载自定义数据集 #1126

Open yanchaoguo opened 3 years ago

yanchaoguo commented 3 years ago

想在官方提供的图像分割模型中进行迁移学习 ，看到2.0版本支持了paddlehub.finetune.trainer 接口，请问如何加载自定义数据集进行训练呢

yanchaoguo commented 3 years ago

u2net该模型支持自定义数据集训练吗

haoyuying commented 3 years ago

自定义数据集可查看https://github.com/PaddlePaddle/PaddleHub/blob/release/v2.0.0-beta/docs/docs_ch/tutorial/how_to_load_data.md u2net目前不支持自定义数据集训练，源码可参见https://github.com/PaddlePaddle/PaddleHub/blob/release/v2.0.0-beta/modules/image/semantic_segmentation/U2Net/module.py

yanchaoguo commented 3 years ago

自定义数据集可查看https://github.com/PaddlePaddle/PaddleHub/blob/release/v2.0.0-beta/docs/docs_ch/tutorial/how_to_load_data.md u2net目前不支持自定义数据集训练，源码可参见https://github.com/PaddlePaddle/PaddleHub/blob/release/v2.0.0-beta/modules/image/semantic_segmentation/U2Net/module.py

是的 源码看过了 paddlehub2.0 不是提供了高级版接口 支持finetune吗，

haoyuying commented 3 years ago

支持finetune的模型需要额外增加代码支持，可以参考https://github.com/PaddlePaddle/PaddleHub/blob/release/v2.0.0-beta/paddlehub/module/cv_module.py 这部分代码就是支持finetune的模型额外需要增加的部分。

Fitz-Fitz commented 3 years ago

这种情况有提供对应的demo吗