search

ajaichemmanam / simple_bodypix_python

A simple and minimal bodypix inference in python
76 stars 19 forks source link

Error using mobilenet instead of resnet50 #7

Closed RazZziel closed 4 years ago

RazZziel commented 4 years ago

This is probably a newbie issue, I'm sorry if the solution is obvious!

The code works fine with the network bodypix_resnet50_float_model-stride16

In that network, segmentationMask.shape has the following value: (76, 101, 1)

However, the code fails with the networks bodypix_mobilenet_quant2_075_model-stride16 and bodypix_mobilenet_float_075_model-stride16, with the following error: ValueError: cannot reshape array of size 245632 into shape (76,101)

In that network, segmentationMask.shape has the following value: (76, 101, 32)

RazZziel commented 4 years ago

Nevermind, I just realised you were just talking about that in https://github.com/ajaichemmanam/simple_bodypix_python/issues/4 and that it's already fixed, thanks!