Ошибка несовместимых форм при обучении

Question

Я тренировал детектор объектов, смог успешно сделать это с 1 классом, теперь я добавил 2 класса и еще несколько изображений в набор данных, я восстановил xml в csv и tf записи, удалив старые,Файл pbtxt правильный, а также удаляет старые контрольные точки в каталоге обучения, но когда я запускаю учебное задание, как это:

python legacy/train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

после этого учебного пособия .Я получаю следующую ошибку:

InvalidArgumentError (see above for traceback): Incompatible shapes: [2,1917] vs. [4,1]
 [[Node: Loss/Match/cond/mul_4 = Mul[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Loss/Match/cond/one_hot, Loss/Match/cond/Cast_2)]]

Answer 1

Если вы обучаете SSD Mobilenet, следуя учебному пособию по API обнаружения препятствий, вы можете попробовать закомментировать дополнение "ssd_random_crop" в вашем конфигурационном файле.Это сработало для меня.Первоначально был отмечен rky0930 здесь .