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 do I train my own data set? #21

Open
Joey1224 opened this issue Mar 14, 2024 · 1 comment
Open

How do I train my own data set? #21

Joey1224 opened this issue Mar 14, 2024 · 1 comment

Comments

@Joey1224
Copy link

HI~ Thank you for your work!
I want to train a model that recognizes cars and boats, is the train config and CSV file labeled correctly?
config
csv

@IceClear
Copy link
Owner

IceClear commented Apr 1, 2024

Hi. Our model is for image quality assessment rather than classification. So I am not sure if it is suitable to directly training it for your case. Our output is multiplied by 100. So you may need to rescale it to [0,1] before calculating the loss.

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