liangjiandeng / DLPan-Toolbox

DLPan Toolbox for Pansharpening
GNU General Public License v3.0
138 stars 19 forks source link

能否同时读取两个数据集并进行联合训练? #14

Open stxg13056 opened 1 year ago

stxg13056 commented 1 year ago

我为自己的模型加入了联合训练的功能,但是我没有找到很好的方法将h5格式存储的两种数据集同时读入模型(只能按顺序训练完一个再训练另一个)。我尝试了在训练之前就把两个数据集进行合并,但我发现h5文件中,每个子数据集(‘pan’,‘lms’,‘ms’,‘gt’)都是采用四维张量的形式保存的,无法像两个不同的文件夹一样堆叠在一起。因为在将两个数据集的内容拼接在一起的时候,会出现其他维度的不匹配。(我希望混合4个波段和8个波段的数据集来进行联合训练)