search

Zehaos / MobileNet

MobileNet build with Tensorflow
Apache License 2.0
1.62k stars 470 forks source link

MobileNet does not take input in batches #61

Open sinbycos opened 6 years ago

sinbycos commented 6 years ago

Using the Dataset Tensorflow API I am providing input to MobileNet, but it cannot take input in batches. How can I get predicitons from the model batch wise.

sinbycos commented 6 years ago

I am trying to use insert_placeholder.py script. And I am using the checkpoint files from https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet_v1.md

However the script gives me the following error:

NotFoundError (see above for traceback): Key MobileNet/conv_1/batch_norm/moving_variance not found in checkpoint
     [[Node: save/RestoreV2_2 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2_2/tensor_names, save/RestoreV2_2/shape_and_slices)]]
     [[Node: save/RestoreV2_27/_59 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name="edge_336_save/RestoreV2_27", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"]()]]