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

When the shape of input is [1,3,112,96], the shape of output is [2,10574]? #54

Open
yBeOne opened this issue Jun 10, 2020 · 4 comments
Open

Comments

@yBeOne
Copy link

yBeOne commented Jun 10, 2020

A torch.randn[1,3,112,96] is as the input ,but the output is shape of [2,10574]? Why not [1,10574]?

@njuaplusplus
Copy link

The output is a tuple, according to the code.

But I don't know how to use this repo as a classifier.

@clcarwin
Copy link
Owner

clcarwin commented Feb 5, 2021

please read lfw_eval.py: set net.feature = True for eval.
fc5 is the embedding, fc6 is only used for train.

@njuaplusplus
Copy link

Thank you for your reply.
I understand that for face verification (compute the distances), we need to set net.feature=True.
My question is if I want to use the fc6 so the network functions as a classifier, what should I do?
I think it's reasonable since it's trained as a classifier.

@njuaplusplus
Copy link

More specifically, could you give me some hints on how to get the prediction confidence of a given image with your pre-trained network.
Thanks a lot.

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