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

Training #127

Open
luciaL opened this issue May 9, 2017 · 3 comments
Open

Training #127

luciaL opened this issue May 9, 2017 · 3 comments

Comments

@luciaL
Copy link

luciaL commented May 9, 2017

I trained my own data using TVG_new_traintest.prototxt. I changed the number of output and the name of the deconvolution layer as well as initialize the weights of all layers.But the output is still all black.Then I visualized feature of every layer.I found that the feature become all black after the first deconvolution layer.Do you know where the problem lie???

@bittnt
Copy link
Collaborator

bittnt commented May 10, 2017

This is because you did not successfully initialize the deconv layer.

@Sam813
Copy link

Sam813 commented May 31, 2018

@luciaL I have the same issue, may I know how did you solve it?

below is the way that I did it,
since My images consist of two class of background and tumors I changed the 21 to 2, and also renamed it to xxx_new

layer {
name: "score2-new"
type: "Deconvolution"
bottom: "score"
top: "score2-new"
param {
lr_mult: 1
}
convolution_param {
num_output: 2 #21
kernel_size: 4
stride: 2
weight_filler: { type: "xavier"
std: 0.1
}
bias_filler {
type: "constant"
value: 0.2 }
}
}
So as you see, i changed the number of outputs to 2 and initialize the weights by xavier method.

I have included my train prototext in case it can help.

Train_prototext.txt

@luciaL
Copy link
Author

luciaL commented Jun 5, 2018

Thank you so much @Sam813 . I have finished this work already. Good luck.

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

3 participants