Pytorch للتجزئة الدلالية
يحتوي هذا المستودع على بعض النماذج للتجزئة الدلالية وخط أنابيب نماذج التدريب والاختبار ، والتي تم تنفيذها في Pytorch
النماذج
- Vanilla FCN: FCN32 ، FCN16 ، FCN8 ، في إصدارات VGG و RESNET و Densenet على التوالي (شبكات تلافيفية بالكامل للتجزئة الدلالية)
- U-Net (U-Net: شبكات تلافيفية لتجزئة الصور الطبية الحيوية)
- SEGNET (SEGNET: بنية ترفيهية ترفيهية عميقة لتجزئة الصور)
- PSPNET (شبكة تحليل مشهد الهرم)
- GCN (مسائل النواة الكبيرة)
- DUC ، HDC (فهم الالتفاف للتجزئة الدلالية)
متطلبات
- Pytorch 0.2.0
- Tensorboard ل Pytorch. هنا للتثبيت
- بعض المكتبات الأخرى (ابحث عن ما تفوتك عند تشغيل الكود: p)
تحضير
- انتقل إلى دليل النماذج وضبط مسار النماذج المسبق في config.py
- انتقل إلى دليل مجموعات البيانات والقيام باتباع ReadMe
تودو
- Deeplab V3
- refinenet
- المزيد من مجموعة البيانات (على سبيل المثال)