search

kentaroy47 / frcnn-from-scratch-with-keras

:collision:Faster R-CNN from scratch written with Keras
Apache License 2.0
168 stars 106 forks source link

Batch Size #41

Open syifalaili26 opened 5 years ago

syifalaili26 commented 5 years ago

Is Faster RCNN using batch_size when training no or epoch_length is batch_size? Can you please explain it, I can't find the batch_size?

lijingpeng commented 4 years ago

batch_size = 1. you can find in the code https://github.com/kentaroy47/frcnn-from-scratch-with-keras/blob/3511bb2cc92b785622441a34469be118d5c1d6a0/keras_frcnn/data_generators.py#L338

so epoch_length should be the size of the train dataset

Mps24-7uk commented 4 years ago

@lijingpeng can we change batch_size = 2