microsoft / MaskFlownet

[CVPR 2020, Oral] MaskFlownet: Asymmetric Feature Matching with Learnable Occlusion Mask

Home Page:https://arxiv.org/abs/2003.10955

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How can I infer my own data by net MaskFlownet_S?

Yugt4732 opened this issue · comments

AssertionError: Parameter 'maskflownet_s0_hybridsequential0_conv1aweight' is missing in file './MaskFlownet/weights/dbbSep30-1206_1000000.params', which contains parameters: 'maskflownet0_hybridsequential0_conv1aweight', 'maskflownet0_hybridsequential0_conv1abias', 'maskflownet0_hybridsequential1_conv1bweight', ..., 'maskflownet0_hybridsequential55_conv3fweight', 'maskflownet0_hybridsequential55_conv3fbias', 'maskflownet0_hybridsequential56_conv2fweight', 'maskflownet0_hybridsequential56_conv2fbias'. Please make sure source and target networks have the same prefix.

I get error as above, and how can I get occlusion masks as paper's work?
this is the command I used:
python predict_new_data.py ./test.png MaskFlownet_S.yaml --image_1 ./image_1.png --image_2 ./image_2.png -g 0 -c 5adNov03