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].
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 https://github.com/BichenWuUCB/squeezeDet/issues/4#issuecomment-552423891