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

train model on own data? #2

Open
xiaoxiao079 opened this issue Jan 23, 2024 · 1 comment
Open

train model on own data? #2

xiaoxiao079 opened this issue Jan 23, 2024 · 1 comment

Comments

@xiaoxiao079
Copy link

hi there i want to train you model on our images. could it possible to train you model on grey scale images, and what are the changes which would needed to be done before training it on custom data?

@Smith42
Copy link
Owner

Smith42 commented May 1, 2024

Yes you can train on greyscale images, you would need to normalise them with a new dataloader so that they have a scale between -1 and 1. You would also need to change the network input channels to be 1.

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