-
Notifications
You must be signed in to change notification settings - Fork 12
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
object detection how to change input size #4
Comments
I don't think you could change the input dimensions on the fly. You'll have to train a model of 240x240 yourself. |
@YijinLiu |
@zjbaby I don't know about Caffe. Do you mean if you give Caffe a smaller image, inference will take less time? Tensorflow SSD model does support resizing the input. It won't affect the inference speed since what it does is essentially resizing the image.. |
@YijinLiu |
@YijinLiu |
@YijinLiu @camect
Hi , I test mobilenet ssd model which downloaded from
"https://storage.googleapis.com/download.tensorflow.org/models/tflite/coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.zip"
and I try to change net input size for faster speed by the following code:
`
but i got an error:
resize input tensor: ( 1 240 240 3 )
tensorflow/contrib/lite/kernels/reshape.cc:58 num_input_elements != num_output_elements (2700 != 4332)
Node number 36 (RESHAPE) failed to prepare.
Failed to allocate tensors!
Am I wrong? what should I do if I want to change net input size arbitrarily?
The text was updated successfully, but these errors were encountered: