الصفحة الرئيسية>المتعلقة بالبرمجة>كود الذكاء الاصطناعي
تنزيل

الشبكة العصبية التلافيفية ذات الشكل العلاقة لتحليل سحابة النقاط

يحتوي هذا المستودع على تنفيذ المؤلف في Pytorch للورقة:

الشبكة العصبية التلافيفية ذات الشكل العلاقة لتحليل سحابة النقاط [arXiv] [CVF]
يونغتشنغ ليو، بن فان، شيمينغ شيانغ وتشونهونغ بان
صفحة مشروع CVPR 2019 الشفهية وأفضل ورقة نهائية : https://yochengliu.github.io/Relation-Shape-CNN/

الاقتباس

إذا كانت ورقتنا مفيدة لبحثك، يرجى النظر في الاستشهاد بما يلي: 

        @inproceedings { liu2019rscnn ,   
            author = { Yongcheng Liu and    
                            Bin Fan and    
                      Shiming Xiang and   
                           Chunhong Pan } ,   
            title = { Relation-Shape Convolutional Neural Network for Point Cloud Analysis } ,   
            booktitle = { IEEE Conference on Computer Vision and Pattern Recognition (CVPR) } ,    
            pages = { 8895--8904 } ,  
            year = { 2019 }   
        }

الاستعمال: التحضير

متطلبات

  • أوبونتو 14.04
  • بايثون 3 (يوصي بـ Anaconda3)
  • بيتورتش 0.3.*/0.4.*
  • كميك> 2.8
  • كودا 8.0 + كيو دي إن إن 5.1

بناء النواة 

 git clone https://github.com/Yochengliu/Relation-Shape-CNN.git 
cd Relation-Shape-CNN
  • بناء mkdir && بناء القرص المضغوط
  • cmake .. && make

مجموعة البيانات

تصنيف الشكل

قم بتنزيل وفك ضغط ModelNet40 (415M). استبدل $data_root$ في cfgs/config_*_cls.yaml بالمسار الأصلي لمجموعة البيانات.

تجزئة جزء ShapeNet

قم بتنزيل وفك ضغط ShapeNet Part (674M). استبدل $data_root$ في cfgs/config_*_partseg.yaml بمسار مجموعة البيانات.

الاستخدام: التدريب

تصنيف الشكل 

 sh train_cls.sh

يمكنك تعديل relation_prior في cfgs/config_*_cls.yaml . لقد قمنا بتدريب نموذج تصنيف حي أحادي النطاق في مجلد cls ، والذي تبلغ دقته 92.38%.

تجزئة جزء الشكل 

 sh train_partseg.sh

لقد قمنا بتدريب نموذج تجزئة الأجزاء متعدد المقاييس في مجلد seg ، الذي تبلغ نسبة mIoU الخاصة بفئته ومثيل mIoU 84.18% و85.81% على التوالي.

الاستخدام: التقييم

تصنيف الشكل 

 Voting script: voting_evaluate_cls.py

يمكنك استخدام نموذجنا cls/model_cls_ssn_iter_16218_acc_0.923825.pth كنقطة تفتيش في config_ssn_cls.yaml ، وبعد هذا التصويت ستحصل على دقة تبلغ 92.71% إذا سارت الأمور على ما يرام.

تجزئة جزء الشكل 

 Voting script: voting_evaluate_partseg.py

يمكنك استخدام نموذجنا seg/model_seg_msn_iter_57585_ins_0.858054_cls_0.841787.pth كنقطة تفتيش في config_msn_partseg.yaml .

رخصة

تم إصدار الكود بموجب ترخيص MIT (راجع ملف الترخيص للحصول على التفاصيل).

شكر وتقدير

تم استعارة الكود بشكل كبير من Pointnet2_PyTorch.

اتصال

إذا كانت لديك بعض الأفكار أو الأسئلة حول أبحاثنا لمشاركتها معنا، فيرجى الاتصال بـ [email protected]

يوسع
معلومات إضافية
تطبيقات ذات صلة
نوصي لك
أخبار ذات صلة الكل