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

Reasoning behind output shape #20

Open
schoennenbeck opened this issue Aug 9, 2021 · 0 comments
Open

Reasoning behind output shape #20

schoennenbeck opened this issue Aug 9, 2021 · 0 comments

Comments

@schoennenbeck
Copy link

Is there a reason why the output (both for the column and the table mask) has 3 channels? The training is basically doing binary classification (which also means the mask does not need to be of dtype float32, but that's another matter) so we could work with either 2 output channels (those being the logits for "belongs to the table/column or not") or even just one channel (which modulo applying the sigmoid-function would give the probability that a pixel belongs to the table/column).

Since the labels contain only zeros and ones the third channel should get arbitrarily small values after training long enough anyway.

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

1 participant