Keras中的多输入ImageDataGenerator图片生成器

最新推荐文章于 2024-11-17 13:18:33 发布

明年暑假升初中

最新推荐文章于 2024-11-17 13:18:33 发布

阅读量4.3k

点赞数

分类专栏：机器学习

本文链接：https://blog.csdn.net/yjy728/article/details/79571138

版权

参考keras官网以及https://github.com/Deep-Learning-Person-Re-Identification/
通过重载Keras自带的ImageDataGenerator来实现同时输入多张图片。该代码是为了实现Keras中的TripletNet中的Triplet Loss。
更多部分参考：
http://blog.csdn.net/yjy728/article/details/79570554
http://blog.csdn.net/yjy728/article/details/79569807
代码环境：

keras：2.1.2
tensorflow：1.4.0
python3.6
win7

from keras import backend as K
import numpy as np
from PIL import Image
from keras.preprocessing.image import ImageDataGenerator, Iterator
from keras.utils import np_utils
class ImageDataGenerator_Triplet(ImageDataGenerator):
    def flow(self, basepath, batch_size=32, class_num=731, input_size=299,
             train_vali_flag = 'train',
             shuffle=False, seed=None,
             save_to_dir=