Referentie voor ultralytics/models/sam/amg.py
Opmerking
Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/amg .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Geeft een booleaanse tensor terug die aangeeft of de vakken zich dicht bij de bijsnijdrand bevinden.
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Verzamel batches van gegevens uit de invoerargumenten.
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Berekent de stabiliteitsscore voor een batch maskers.
De stabiliteitsscore is de IoU tussen de binaire maskers die wordt verkregen door het drempelen van de voorspelde maskerlogits bij hoge en lage waarden.
Opmerkingen
- Het ene masker zit altijd in het andere.
- Bespaar geheugen door onnodige cast naar torch.int64 te voorkomen
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
Genereer een 2D raster van gelijkmatig verdeelde punten in het bereik [0,1]x[0,1].
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Genereer puntrasters voor alle gewaslagen.
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Genereert een lijst met uitsnijdboxen van verschillende groottes.
Elke laag heeft (2i)2 vakjes voor de i-de laag.
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Ontsnijd bounding boxes door de crop box offset toe te voegen.
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Ontsnijd punten door de offset van het uitsnijdvak toe te voegen.
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
Ontcrop maskers door ze op te vullen tot de originele afbeeldingsgrootte.
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Kleine losgekoppelde gebieden of gaten in een masker verwijderen, waarbij het masker en een wijzigingsindicator terugkomen.
Broncode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Berekent vakken in XYXY-formaat rond maskers.
Geeft [0,0,0,0] terug voor een leeg masker. Voor de invoervorm C1xC2x...xHxW is de uitvoervorm C1xC2x...x4.