Python 当图像';s像素和(多个)y标签保存在两个单独的数据帧中
我正在CNN上使用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
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