FSL-Mate icon indicating copy to clipboard operation
FSL-Mate copied to clipboard

您好,是否可以介绍下自定义的数据集如何实现训练评估等操作

Open lmqhello opened this issue 1 year ago • 2 comments

最好能够详细一些,非常感谢!

lmqhello avatar Oct 24 '23 16:10 lmqhello

是否是 TRAIN_DATASET = paddlefsl.datasets.FC100(mode='train') VALID_DATASET = paddlefsl.datasets.FC100(mode='valid') TEST_DATASET = paddlefsl.datasets.FC100(mode='test')

把FC100修改成

lmqhello avatar Oct 27 '23 19:10 lmqhello

@lmqhello 是的,但是需要做一些小的修改,步骤如下:

  1. 自定义一个数据预处理类,参照/paddlefsl/datasets下的方法:https://github.com/tata1661/FSL-Mate/tree/master/PaddleFSL/paddlefsl/datasets 中的方法(如fc100中是通过file_name来指定FC数据集),如果是自定义数据集可以将文件校验部分改为pass 2.自定义一个训练类:参照examples中的方法,如(examples/image_classification/relationnet_image_classification.py, 将TRAIN_DATASET = paddlefsl.datasets.FC100换成自定义的数据预处理类

Ren-Xin avatar Mar 26 '24 12:03 Ren-Xin

@Ren-Xin thanks~

tata1661 avatar May 17 '24 05:05 tata1661