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

How to train & test on big image #123

Closed
Oktai15 opened this issue Mar 24, 2018 · 4 comments
Closed

How to train & test on big image #123

Oktai15 opened this issue Mar 24, 2018 · 4 comments

Comments

@Oktai15
Copy link

Oktai15 commented Mar 24, 2018

Hello, @argman and @zxytim !

Thank you for your job it's really cool. I need your advice.

I have a lot of image that have size about 4000x3000 and a lot of small text bbox's on each image.
What should I do on training? I see two ways:

  1. directly pass this size and increase max_side_length & decrease text_scale;
  2. cut big image into pieces (but text can be splitted)

And what should I do on testing? Again two ways:

  1. directly pass this size and increase max_side_length & decrease text_scale;
  2. cut big image into pieces (here I can use NMS)

What would you choose? Or maybe something else?

Btw, do you support multi-scale training?

Thanks in advance!

@argman
Copy link
Owner

argman commented Mar 25, 2018

I would choose the first one, if it cannot fit into gpu-memory ,then you can downsize a bit or train a smaller network

@Oktai15
Copy link
Author

Oktai15 commented Mar 25, 2018

@argman, thanks for answer!
You skipped one question about multi-scale training)

@argman
Copy link
Owner

argman commented Mar 25, 2018

I have already cut pieces during training(but ensure text will not be splitted), tuning of parameters may help, it depends on your situation

@Oktai15
Copy link
Author

Oktai15 commented Mar 25, 2018

@argman , okay, my last questions:
Could you point at code where you ensure that bounding boxes won’t be splitted?
Do you restore right coordinates on testing to use NMS and correct problem with splitted texts?

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