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

Fine Tuning on our custom dataset #91

Open
Arnab1181412 opened this issue Jan 15, 2024 · 5 comments
Open

Fine Tuning on our custom dataset #91

Arnab1181412 opened this issue Jan 15, 2024 · 5 comments

Comments

@Arnab1181412
Copy link

Hi I have image pairs for my own data and I want to fine tune using this dataset. There are around 500 images of 255 X 255 size. How do I do it?
Thanks in advance

@MengyangPu
Copy link
Owner

MengyangPu commented Jan 16, 2024

@Arnab1181412
The process is the same as the training process:
1 Data augment, and generate a train_pair.txt of your data.
2 Modify the path in the configuration file.
3 Adjust parameters, e.g., learning rate, and batch size.

@Arnab1181412
Copy link
Author

Arnab1181412 commented Jan 16, 2024 via email

@MengyangPu
Copy link
Owner

Yes, and you could create your dataset.py based on configs/base/datasets/bsds.py.

@Arnab1181412
Copy link
Author

Arnab1181412 commented Jan 16, 2024 via email

@Arnab1181412
Copy link
Author

Arnab1181412 commented Jan 17, 2024 via email

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