You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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 vectorout_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.
The text was updated successfully, but these errors were encountered: