Darkflow: टेंसर फ्लो टेंसर रीशेप एरर

मैं योलो को अपने डेटासेट पर प्रशिक्षित करने की कोशिश कर रहा हूं। मैं 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

वास्तविक त्रुटि:

op 'Reshape' के कारण, यहां परिभाषित:
फ़ाइल "./flow", लाइन 44, in
टीएफनेट = टीएफनेट (झंडे)
फ़ाइल "/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 में
सेल्फ.फ्रेमवर्क.लॉस (सेल्फ.आउट)
फ़ाइल "/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 में
ओरिजिनल_ऑप = सेल्फ._डिफॉल्ट_ओरिजिनल_ऑप, op_def=op_def)
फ़ाइल "/home/ubuntu/.conda/envs/py3/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", लाइन 1226, __init__ में
स्व._ट्रेसबैक = _extract_stack ()
InvalidArgumentError (ट्रेसबैक के लिए ऊपर देखें): पुन: आकार देने के लिए इनपुट 1149200 मानों वाला एक टेंसर है, लेकिन अनुरोधित आकार के लिए 18590 के गुणक की आवश्यकता होती है
[[नोड: रीशेप = रीशेप [टी = डीटी_एफएलओएटी, टीशैप = डीटी_आईएनटी 32, _ डिवाइस = "/ जॉब: लोकलहोस्ट/प्रतिकृति : 0/ कार्य: 0/जीपीयू : 0"] (आउटपुट, रीशेप/आकार)]]
[[नोड: mul_30/_195 = _Recv [क्लाइंट_टर्मिनेटेड = झूठा, recv_device = "/ जॉब: लोकलहोस्ट/प्रतिकृति : 0/ कार्य: 0/सीपीयू : 0", send_device = "/ जॉब: लोकलहोस्ट/प्रतिकृति : 0/ कार्य: 0 /gpu :0", send_device_incarnation=1, tensor_name="edge_188_mul_30", tensor_type=DT_FLOAT, _device="/ job:localhost/ replication: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 , @ ध्रुव-मोहन, @abagshaw क्या आपके पास कोई विचार है कि मुझे यह त्रुटि क्यों मिल रही है?

सहायता के लिए धनयवाद