Darkflow: 텐서 흐름 텐서 재구성 오류

내 데이터 세트에서 yolo를 훈련시키려고 합니다. 나는 2개의 객체를 가진 10개의 이미지에 성공적으로 기차를 과적합할 수 있었습니다. 추론이 잘 작동합니다.

이제 6000개의 이미지로 구성된 전체 데이터 세트를 학습하려고 하는데 오류가 발생합니다.
훈련에 사용하는 cmd:
./flow --train --dataset /home/ubuntu/datasets/img/ --annotation /home/ubuntu/datasets/anno/ --model cfg/yolo-17c.cfg --load bin/yolo.weights --keep 5 --epoch 30000 --save 1000 --lr 0.00001 --batch 16 --gpu .8

실제 오류:

다음에서 정의된 작업 'Reshape'로 인해 발생:
파일 "./flow", 44행,
tfnet = TFNet(FLAGS)
파일 "/home/ubuntu/darkflow/net/build.py", 63행, __init__
self.setup_meta_ops()
setup_meta_ops의 파일 "/home/ubuntu/darkflow/net/build.py", 106행
if self.FLAGS.train: self.build_train_op()
build_train_op에서 파일 "/home/ubuntu/darkflow/net/help.py", 15행
self.framework.loss(self.out)
파일 "/home/ubuntu/darkflow/net/yolov2/train.py", 56행, 손실
net_out_reshape = tf.reshape(net_out, [-1, H, W, B, (4 + 1 + C)])
파일 "/home/ubuntu/.conda/envs/py3/lib/python3.5/site-packages/tensorflow/python/ops/gen_array_ops.py", 라인 2630, 모양 변경
이름=이름)
파일 "/home/ubuntu/.conda/envs/py3/lib/python3.5/site-packages/tensorflow/python/framework/op_def_library.py", 763행, apply_op
op_def=op_def)
파일 "/home/ubuntu/.conda/envs/py3/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", 2327행, create_op
original_op=self._default_original_op, op_def=op_def)
파일 "/home/ubuntu/.conda/envs/py3/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", 1226행, __init__
self._traceback = _extract_stack()
InvalidArgumentError(위의 역추적 참조): 모양 변경에 대한 입력은 1149200 값을 가진 텐서이지만 요청한 모양에는 18590의 배수가 필요합니다.
[[노드: Reshape = Reshape[T=DT_FLOAT, Tshape=DT_INT32, _device="/ job:localhost/replica :0/ task:0/gpu :0"](출력, Reshape/shape)]]
[[노드: mul_30/_195 = _Recv[client_terminated=false, recv_device="/ job:localhost/replica :0/ task:0/cpu :0", send_device="/ job:localhost/replica :0/ task:0 /gpu :0", send_device_incarnation=1, tensor_name="edge_188_mul_30", tensor_type=DT_FLOAT, _device="/ 작업:localhost/복제본 :0/ 작업:0/cpu :0"]()]]

내 labels.txt 파일에는 17개의 레이블이 있습니다.
yolo-17c.cfg에 대한 변경 사항은 다음과 같습니다.
filters=125 마지막 레이어에 대해 425 및 35 로도 시도했습니다.
classes=17 수업 수 기준

18590 숫자를 width=224, height=224 변경할 때 width=416, height=416 와 연관되어 있다는 것을 깨달았습니다.
...but the requested shape requires a multiple of 21560

@thtrieu , @Dhruv-Mohan , @abagshaw 왜 이 오류가 발생하는지 아십니까?

도와 줘서 고마워