Darkflow: خطأ في إعادة تشكيل موتر تدفق الموتر

أحاول تدريب yolo على مجموعة البيانات الخاصة بي. تمكنت من زيادة التدريب بنجاح على 10 صور مع كائنين. الاستنتاج يعمل بشكل جيد.

الآن أحاول التدرب على مجموعة البيانات الكاملة المكونة من 6000 صورة وأحصل على خطأ.
كمد الذي أستخدمه للتدريب:
./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" المحدد في:
ملف ". / تدفق" ، السطر 44 ، في
tfnet = TFNet (أعلام)
ملف "/home/ubuntu/darkflow/net/build.py" ، السطر 63 ، في __init__
self.setup_meta_ops ()
ملف "/home/ubuntu/darkflow/net/build.py" ، السطر 106 ، في setup_meta_ops
إذا كان self.FLAGS.train: self.build_train_op ()
ملف "/home/ubuntu/darkflow/net/help.py" ، السطر 15 ، في build_train_op
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 ، في application_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 (انظر أعلاه للحصول على traceback): الإدخال لإعادة التشكيل هو موتر بقيم 1149200 ، لكن الشكل المطلوب يتطلب مضاعف 18590
[[Node: Reshape = Reshape [T = DT_FLOAT، Tshape = DT_INT32، _device = "/ job: localhost / replica : 0 / task: 0 / gpu : 0"] (الإخراج ، إعادة الشكل / الشكل)]]
[[العقدة: 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 =" / job: localhost / replica : 0 / task: 0 / cpu : 0 "] ()]]

يحتوي ملف labels.txt على 17 تصنيفًا.
التغييرات التي أجريتها على yolo-17c.cfg هي:
filters=125 للطبقة الأخيرة لقد جربت هذا أيضًا باستخدام 425 و 35
classes=17 بناءً على عدد الفصول

أدرك أن 18590 مرتبط بـ width=416, height=416 عندما أقوم بتغييرهم إلى width=224, height=224 ، أحصل على
...but the requested shape requires a multiple of 21560

thtrieu ، @ Dhruv-Mohan ، abagshaw هل لديك أي أفكار لماذا أتلقى هذا الخطأ؟

شكرا للمساعدة