Detectron: خطأ التوكيد:

تم إنشاؤها على ٢٦ يوليو ٢٠١٨ · 3تعليقات · مصدر: facebookresearch/Detectron

نتائج متوقعة

ادارة

النتائج الفعلية

كتلة مساحة العمل cls_score_w بالشكل (21 ، 2048) لا تتطابق مع شكل ملف الأوزان (81 ، 2048)

خطوات مفصلة لإعادة الإنتاج

على سبيل المثال:

 python tools/train_net.py --cfg /root/Detectron/tmp/e2e_faster_rcnn_R-50-C4_1x.yaml OUTPUT_DIR tmp/det

معلومات النظام

نظام التشغيل: أوبونتو
إصدار المترجم: caffe2
إصدار CUDA: 9
إصدار cuDNN: 7
إصدار برنامج تشغيل NVIDIA: Titanx
طرز GPU (لجميع الأجهزة إذا لم تكن كلها متشابهة):؟
PYTHONPATH متغير البيئة: / root / anaconda
الناتج python --version : 2.7
أي شيء آخر يبدو ذا صلة:؟

مصدر

elfpattern

التعليق الأكثر فائدة

من المحتمل أنك قمت بتعيين NUM_CLASSES: 21 في ملف التكوين الخاص بك ، بينما يحتوي ملف الأوزان على 81 فئة. طريقة واحدة للتغلب على ذلك هو مجرد تحميله وحذف النقط المتضاربة.

import cPickle as pkl 

with open('model.pkl', 'rb') as f:
    wts = pkl.load(f)

for blob in wts['blobs'].keys():
    if blob.startswith('cls_score_') or blob.startswith('bbox_pred_'):
        del wts['blobs'][blob]

with open('new_model.pkl', 'wb') as f:
    pkl.dump(wts, f)

أتمنى أن يساعدك هذا.

zeenolife في ٢٧ يوليو ٢٠١٨

👍5

ال 3 كومينتر

import cPickle as pkl 

with open('model.pkl', 'rb') as f:
    wts = pkl.load(f)

for blob in wts['blobs'].keys():
    if blob.startswith('cls_score_') or blob.startswith('bbox_pred_'):
        del wts['blobs'][blob]

with open('new_model.pkl', 'wb') as f:
    pkl.dump(wts, f)

أتمنى أن يساعدك هذا.

zeenolife في ٢٧ يوليو ٢٠١٨

👍5

لقد فعلتها thx

elfpattern في ٢٧ يوليو ٢٠١٨

👍1

من المحتمل أنك قمت بتعيين NUM_CLASSES: 21 في ملف التكوين الخاص بك ، بينما يحتوي ملف الأوزان على 81 فئة. طريقة واحدة للتغلب على ذلك هو مجرد تحميله وحذف النقط المتضاربة.
import cPickle as pkl 

with open('model.pkl', 'rb') as f:
    wts = pkl.load(f)

for blob in wts['blobs'].keys():
    if blob.startswith('cls_score_') or blob.startswith('bbox_pred_'):
        del wts['blobs'][blob]

with open('new_model.pkl', 'wb') as f:
    pkl.dump(wts, f)
أتمنى أن يساعدك هذا.

في أي ملف أحتاج إلى إضافة هذه التفاصيل