search

omni-us / squeezedet-keras

Keras implementation of the Squeeze Det Object Detection Deep Learning Framework
MIT License
129 stars 41 forks source link

Network output does not match? #14

Closed kentaroy47 closed 6 years ago

kentaroy47 commented 6 years ago

Great work, and I want to try this out myself.. However, when I compile the model, I get such errors.

The shape of preds is preds: (?, 72, 78, 256) and anchors: 16848 so it cannot reshape.

I was wondering how people is getting around this, or did I install it properly?

File "../../scripts/train.py", line 314, in train() File "../../scripts/train.py", line 131, in train squeeze = SqueezeDet(cfg) File "/home/ken/rev7_ken/keras/objectdetection/squeezedet-keras/main/model/squeezeDet.py", line 32, in init self.model = self._create_model() File "/home/ken/rev7_ken/keras/objectdetection/squeezedet-keras/main/model/squeezeDet.py", line 95, in _create_model pred_reshaped = Reshape((self.config.ANCHORS, -1))(preds) File "/home/ken/miniconda3/lib/python3.6/site-packages/keras/engine/base_layer.py", line 474, in call output_shape = self.compute_output_shape(input_shape) File "/home/ken/miniconda3/lib/python3.6/site-packages/keras/layers/core.py", line 394, in compute_output_shape input_shape[1:], self.target_shape) File "/home/ken/miniconda3/lib/python3.6/site-packages/keras/layers/core.py", line 379, in _fix_unknown_dimension raise ValueError(msg) ValueError: total size of new array must be unchanged

kentaroy47 commented 6 years ago

sorry, the keras configuration was "channel_first" instead of "channel_last". after fixing that issue, the code worked.