yolov3代码涉及到的Keras Model fit_gene

2019-08-17  本文已影响0人  dc_3

fit_generator

使用 Python 生成器(或 Sequence 实例)逐批生成的数据,按批次训练模型。生成器与模型并行运行,以提高效率。 例如,这可以让你在 CPU 上对图像进行实时数据增强,以在 GPU 上训练模型。keras.utils.Sequence 的使用可以保证数据的顺序, 以及当 use_multiprocessing=True 时 ,保证每个输入在每个 epoch 只使用一次。

fit_generator(generator, steps_per_epoch=None, epochs=1, 
verbose=1, callbacks=None, validation_data=None, 
validation_steps=None, class_weight=None, 
max_queue_size=10, workers=1, use_multiprocessing
=False, shuffle=True, initial_epoch=0)

yolov3->train.py中的model.fit_generator

model.fit_generator(data_generator_wrapper(lines[:num_train]
, batch_size, input_shape, anchors, num_classes),
                steps_per_epoch=max(1, num_train//batch_size),
                validation_data=data_generator_wrapper(lines[num_train:], batch_size, input_shape, anchors, num_classes),
                validation_steps=max(1, num_val//batch_size),
                epochs=50,
                initial_epoch=0,
                callbacks=[logging, checkpoint]) 

在每次调用的时候执行,遇到yield语句返回,再次被调用时候从上次的返回yield语句处急需执行,也就是用多少,取多少,不占内存。

上一篇下一篇

猜你喜欢

热点阅读