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

Weight shape mismatch when running crop_align_vggface2_FFHQalign.py #9

Open
jhkonan opened this issue Dec 20, 2021 · 1 comment
Open

Comments

@jhkonan
Copy link

jhkonan commented Dec 20, 2021

After following the instructions to setup the environment, ./insightface_func/models, and ./experiments/pretrained_models...

Running the following command:

python scripts/crop_align_vggface2_FFHQalign.py --input_dir $DATAPATH$/VGGface2/train --output_dir_ffhqalign $ALIGN_OUTDIR$ --mode ffhq --crop_size 256

Results in the following output

input mean and std: 127.5 127.5
find model: ./insightface_func/models/antelope/glintr100.onnx recognition
find model: ./insightface_func/models/antelope/scrfd_10g_bnkps.onnx detection
set det-size: (320, 320)

  0%|                                                   | 0/500 [00:00<?, ?it/s]2021-12-20 16:21:17.029441668 [W:onnxruntime:, execution_frame.cc:811 VerifyOutputSizes] Expected shape from model of {800,10} does not match actual shape of {200,10} for output 500
2021-12-20 16:21:17.029507982 [W:onnxruntime:, execution_frame.cc:811 VerifyOutputSizes] Expected shape from model of {800,4} does not match actual shape of {200,4} for output 497
2021-12-20 16:21:17.029540840 [W:onnxruntime:, execution_frame.cc:811 VerifyOutputSizes] Expected shape from model of {800,1} does not match actual shape of {200,1} for output 494
2021-12-20 16:21:17.029912492 [W:onnxruntime:, execution_frame.cc:811 VerifyOutputSizes] Expected shape from model of {3200,10} does not match actual shape of {800,10} for output 477
2021-12-20 16:21:17.029973895 [W:onnxruntime:, execution_frame.cc:811 VerifyOutputSizes] Expected shape from model of {3200,4} does not match actual shape of {800,4} for output 474
2021-12-20 16:21:17.030020397 [W:onnxruntime:, execution_frame.cc:811 VerifyOutputSizes] Expected shape from model of {3200,1} does not match actual shape of {800,1} for output 471
...

It continues to throw warnings for hundreds of other weights, which I have not included here.

Is this expected behavior?

@NNNNAI
Copy link
Owner

NNNNAI commented Dec 22, 2021

This is a warning caused by onnxruntime, may be the version of onnxruntime did not match your computer. But it wont affect the effect of the final results.

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