Giters

BichenWuUCB / squeezeDet

A tensorflow implementation for SqueezeDet, a convolutional neural network for object detection.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Image resolution problem

Imkaran opened this issue · comments

commented

Hi Bichen,
I have training set with image resolution of 1280x720. When I modified the image height and width in "kitti_model_config.py" and "kitti_squeezeDet_config.py", I got the following error

ValueError: Cannot reshape a tensor with 648000 elements to shape [10,16848,2] (336960 elements) for 'interpret_output/pred_class_probs' (op: 'Reshape') with input shapes: [324000,2], [3] and with input tensors computed as partial shapes: input[1] = [10,16848,2].

Can you please help me in resolving the issue

Originally posted by @Imkaran in #4 (comment)