thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

ValueError: Dimension 1 in both shapes must be equal #1104

Closed hanouel closed 4 years ago

hanouel commented 4 years ago

Hi, I trained my custom data set on yolo. The training process is successful but when I want to convert checkpoint into a .pb file, this error is appearing: ValueError: Dimension 1 in both shapes must be equal, but are 6 and 7. Shapes are [?,6,6] and [?,7,7]. for 'concat_1' (op: 'ConcatV2') with input shapes: [?,6,6,256], [?,7,7,1024], [] and with computed input tensors: input[2] = <3>. Have you any idea how to fix it?