Python 当图像';s像素和(多个)y标签保存在两个单独的数据帧中

Python 当图像';s像素和(多个)y标签保存在两个单独的数据帧中,python,pandas,keras,deep-learning,conv-neural-network,Python,Pandas,Keras,Deep Learning,Conv Neural Network,我正在CNN上使用keras。我有两个独立的数据帧,其中包含图像的信息片段。其中一个数据帧imgs包含137*236的像素、范围为0-32331的值以及图像的id列 imgs.head(2) >> image_id 0 1 ... 32330 32331 0 Train_50210 246 253 ... 251 250 1 Train_50211 250 245 ... 241 244

我正在CNN上使用
keras
。我有两个独立的
数据帧
,其中包含图像的信息片段。其中一个数据帧
imgs
包含
137*236
的像素、范围为
0-32331
的值以及图像的id列

imgs.head(2)
>>
   image_id       0      1  ...  32330  32331

0  Train_50210  246    253  ...    251    250   
1  Train_50211  250    245  ...    241    244  

第二个数据帧
train
包含图像的标签和每个图像所属的三个不同类的值(例如->将图像想象成植物、鱼、人、动物等生物,它们可以有0条或更多的腿、0条或更多的眼睛和0条或更多的鼻子)

正如您所看到的,
数据帧
都没有排序,所以我考虑了排序和合并,但它们包含成千上万的图像数据

如何获得新的扭曲图像以使模型更坚固?

我查阅了
keras
文档,从
keras
.flow()
都有使用
ImageDataGenerator(此处所有参数)
,但没有给出示例。我知道如何使用目录中的
flow\u,但在我的情况下如何使用
flow()
方法


(或者有更好的替代方法吗?

毫无疑问,这是定制生成器的工作。您可以尝试flow,但我认为按照您的建议加入数据帧是不可避免的在自定义生成器中,您为每个批(内存不足)执行联接,并使用
cv2
albumentations
应用增强。使用
flow
,您可以先加入所有内容,然后将其扔到
flow
@DanielMöller。您可以建议我怎么做吗?因为我对流程没有兴趣。或者你可以给我指任何链接或教程。
train.head(2)
>>

   image_id   class_1   class_2  class_3    

0   Train_5        15        9         5    
1   Train_1       159        0         0