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

ignoring class 0 during training #15

Open
andremakar opened this issue Apr 23, 2024 · 4 comments
Open

ignoring class 0 during training #15

andremakar opened this issue Apr 23, 2024 · 4 comments

Comments

@andremakar
Copy link

Hello, can you tell me if it is possible to exclude the 0th grade from training in your code?

@moonboy12138
Copy link

Of courese you can, just modify the datasetloader and loss criterion file.

@andremakar
Copy link
Author

Moonboy12138 will it in any way affect the quality of learning if I exclude the knowledge that I have in class 0 and leave subjects in class 1, 2, 3, etc.?

Your network takes into account the spatial location of objects, and I exclude them from training, that is, the spatial location of pixels should change.

@andremakar
Copy link
Author

andremakar commented Apr 24, 2024

You also take into account the spatial arrangement of objects on 128x128 windows in semantic segmentation, if I understood correctly.
And if you remove pixels with 0 markup in the data_loader, everything will get mixed up.

@moonboy12138
Copy link

It appears that you only need to modify the ignore_index=255 parameter in line 10 of this file. I understand your concern; removing pixels with a value of 0 could significantly alter the results compared to those reported in the paper.

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