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

Training deeplab on higher resolution images #39

Closed
AbdouMechraoui opened this issue Dec 4, 2021 · 2 comments
Closed

Training deeplab on higher resolution images #39

AbdouMechraoui opened this issue Dec 4, 2021 · 2 comments

Comments

@AbdouMechraoui
Copy link

AbdouMechraoui commented Dec 4, 2021

Hi!

Thanks once again for your responsiveness :)

May I ask about the procedure used to train deeplab-v3 on 1024 images, I noticed the provided code in the repo samples train_interpreter.py, and trains train_deeplab.py on images with 512 x 512 resolution for face_34 task.

When doing the same for 1024 x 1024 images, the cross_validation script results in this:
trained_deeplab_1024_500

Could you provide more information on the procedure for training deeplab on 1024 x 1024 images? How large was the training set that you used? The number of epochs, and batch size? I ask the latter, as I also had some issues with OOM cuda error, when training on 32G NVIDIA Tesla V100.

@arieling
Copy link
Collaborator

arieling commented Dec 8, 2021

Hi. In datasetGAN paper, we used 512 x 512 resolution. We didn't scale it up to 1024* 1024.

@AbdouMechraoui
Copy link
Author

I am closing this, I was able to train deeplab using ~500 images/labels (1024, 1024), 30 epochs, and batch_size=4. Thanks anyways :)

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