https://github.com/felisat/federated-learning/blob/master/data_utils.py

想要在这个联邦学习读取data的文件里，添加一段读取自定义的铁路缺陷检测数据集代码（训练集、测试集和验证集都分为defection和non defection两类），需要怎么改？

下面是别人做的铁路缺陷检测深度学习项目：

https://www.kaggle.com/code/gaya3dev/railtrack-resnet50