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

Purpose of reshaping the training labels #3

Open
kl-31 opened this issue Nov 12, 2018 · 0 comments
Open

Purpose of reshaping the training labels #3

kl-31 opened this issue Nov 12, 2018 · 0 comments

Comments

@kl-31
Copy link

kl-31 commented Nov 12, 2018

I posted a similar question to the original unet repo. Here I'm referring to the following line in the create_train_data method in data.py:
train_label = np.reshape(train_label, (len(txt),self.out_cols * self.out_rows,self.num_class))

What is the purpose of this reshaping that converts the image out_cols, out_rows to a vector out_cols*out_rows? This does not seem to be necessary in the binary case. Is the multi-class case treated differently by the fit_generator method?

Thanks in advance for any insight.

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