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

I want to get the prob the input two picture? #2

Open
mgq1507 opened this issue Jun 13, 2018 · 4 comments
Open

I want to get the prob the input two picture? #2

mgq1507 opened this issue Jun 13, 2018 · 4 comments

Comments

@mgq1507
Copy link

mgq1507 commented Jun 13, 2018

If I change the loss='binary crossentropy' into loss='softmax',I can get the prob?
If you can give me some advice? I'm appreciate it.
Another, I have trained the network but it costs too much time,how long you train?

@mgq1507
Copy link
Author

mgq1507 commented Jun 13, 2018

@anson0910

@anson0910
Copy link
Owner

I think you should be able to get the prob similarly by
prob = self.model.predict([img1, img2])

I recall using GTX 970, it took me around 8 hours to train.

@mgq1507
Copy link
Author

mgq1507 commented Jun 15, 2018

Thank you ,my traing loss curve like this:
1
The curve not like yours.
s_net.train(starting_batch=40000),The trained model store where.
In the following code,the test example, not like yours.
Can you tell me how I can recurrent your result?
Thank you again,@anson0910.

@anson0910
Copy link
Owner

I'm not sure why your results are quite different..
Regarding how to save the model, you can check this link.

Hope it helps, thanks!

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