hollance / BlazeFace-PyTorch

The BlazeFace face detector model implemented in PyTorch
Other
430 stars 91 forks source link

how to use regressor #8

Closed yja1 closed 4 years ago

yja1 commented 4 years ago

weighted = (coordinates * scores).sum(dim=0) / total_score

coordinates * scores error: RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 1