مرجع ل ultralytics/models/sam/amg.py
ملاحظه
هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/نماذج/sam/amg.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
إرجاع قيمة منطقية tensor يشير إلى ما إذا كانت المربعات بالقرب من حافة الاقتصاص.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
احصل على دفعات من البيانات من وسيطات الإدخال.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
يحسب درجة الاستقرار لمجموعة من الأقنعة.
درجة الاستقرار هي IoU بين الأقنعة الثنائية التي تم الحصول عليها عن طريق عتبة سجلات القناع المتوقعة عند أعلى والقيم المنخفضة.
تلاحظ
- يتم احتواء قناع واحد دائما داخل الآخر.
- حفظ الذاكرة عن طريق منع الإرسال غير الضروري إلى torchإنت 64
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
قم بإنشاء شبكة ثنائية الأبعاد من النقاط المتباعدة بالتساوي في النطاق [0,1] x [0,1].
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
إنشاء شبكات نقطية لجميع طبقات الاقتصاص.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
ينشئ قائمة بصناديق الاقتصاص بأحجام مختلفة.
تحتوي كل طبقة على (2i) 2 مربعات لطبقة ith.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
قم بإلغاء اقتصاص المربعات المحيطة بإضافة إزاحة مربع الاقتصاص.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
قم بإلغاء اقتصاص النقاط بإضافة إزاحة مربع الاقتصاص.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
قم بإلغاء اقتصاص الأقنعة عن طريق حشوها إلى حجم الصورة الأصلي.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
قم بإزالة المناطق الصغيرة غير المتصلة أو الثقوب في القناع ، وأعد القناع ومؤشر التعديل.
شفرة المصدر في ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
يحسب المربعات بتنسيق XYXY حول الأقنعة.
قم بإرجاع [0،0،0،0] لقناع فارغ. لإدخال الشكل C1xC2x ... xHxW ، شكل الإخراج هو C1xC2x ... x4.