Referans için ultralytics/models/sam/amg.py
Not
Bu dosya https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/models/ sam/amg .py adresinde mevcuttur. Bir sorun tespit ederseniz lütfen bir Çekme İsteği 🛠️ ile katkıda bulunarak düzeltilmesine yardımcı olun. Teşekkürler 🙏!
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Kutuların kırpma kenarına yakın olup olmadığını gösteren bir boolean tensor döndürür.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Giriş argümanlarından veri yığınları elde edin.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Bir maske grubu için kararlılık puanını hesaplar.
Kararlılık puanı, tahmin edilen maske logitlerinin yüksek değerlerde eşiklenmesiyle elde edilen ikili maskeler arasındaki IoU'dur. ve düşük değerler.
Notlar
- Bir maske her zaman diğerinin içinde bulunur.
- torch.int64 adresine gereksiz dökümleri önleyerek bellekten tasarruf edin
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
0,1]x[0,1] aralığında eşit aralıklı noktalardan oluşan bir 2B ızgara oluşturun.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Tüm ürün katmanları için nokta ızgaraları oluşturun.
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Farklı boyutlarda kırpma kutularının bir listesini oluşturur.
Her katmanda birinci katman için (2i)2 kutu bulunmaktadır.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Kırpma kutusu ofsetini ekleyerek sınırlayıcı kutuları açın.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Kırpma kutusu ofsetini ekleyerek noktaları kırpın.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
Maskeleri orijinal görüntü boyutuna doldurarak kırpın.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Bir maskedeki küçük bağlantısız bölgeleri veya delikleri çıkarın, maskeyi ve bir değişiklik göstergesini geri getirin.
Kaynak kodu ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Maskelerin etrafındaki kutuları XYXY formatında hesaplar.
Boş bir maske için [0,0,0,0] döndürür. C1xC2x...xHxW giriş şekli için çıkış şekli C1xC2x...x4'tür.
Kaynak kodu ultralytics/models/sam/amg.py
Oluşturuldu 2023-11-12, Güncellendi 2024-05-08
Yazarlar: Burhan-Q (1), glenn-jocher (3), Laughing-q (1)