Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Is there any image augmentation operation during the training ? some early-stopping metrics to avoid the network being overfitting ? #29

Closed
a6225301 opened this issue Oct 8, 2018 · 1 comment

Comments

@a6225301
Copy link

a6225301 commented Oct 8, 2018

Hello, Is there any image augmentation operation during the training ?
Meanwhile, any some early-stopping metrics to avoid the network being overfitting ?
Thanks~

@hgaiser
Copy link
Contributor

hgaiser commented Oct 8, 2018

By default, during training, only horizontal flipping is enabled : https://github.com/fizyr/keras-maskrcnn/blob/master/keras_maskrcnn/bin/train.py#L145, but more options are available.

I've never really experienced much overfitting with retinanet / maskrcnn. Are you experiencing overfitting? The problem with Keras and the early stopping callback is that it depends on a metric. I haven't yet managed to change the mAP computations into a Keras metric. Normally I keep an eye on the mAP during training and select the snapshot with the highest mAP values.

@hgaiser hgaiser closed this as completed Oct 8, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants