Referenz fĂŒr ultralytics/models/sam/amg.py
Hinweis
Diese Datei ist verfĂŒgbar unter https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/amg .py. Wenn du ein Problem entdeckst, hilf bitte mit, es zu beheben, indem du einen Pull Request đ ïž einreichst. Vielen Dank đ!
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Gibt einen booleschen Wert tensor zurĂŒck, der angibt, ob sich KĂ€stchen in der NĂ€he der Schnittkante befinden.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Erhalte Datenstapel aus den Eingabeargumenten.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Berechnet die StabilitĂ€tsbewertung fĂŒr einen Stapel von Masken.
Der StabilitĂ€tsscore ist der IoU zwischen den binĂ€ren Masken, die durch Schwellenwerte fĂŒr die vorhergesagten Maskenlogits bei hohen und niedrigen Werten.
Anmerkungen
- Eine Maske ist immer in der anderen enthalten.
- Spart Speicher, indem es unnötiges Casting nach torch.int64 verhindert
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
Erstelle ein 2D-Gitter aus gleichmĂ€Ăig verteilten Punkten im Bereich [0,1]x[0,1].
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Erstelle Punktraster fĂŒr alle Anbaulagen.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Erzeugt eine Liste von Schnittboxen unterschiedlicher GröĂe.
Jede Schicht hat (2i)2 Felder fĂŒr die i-te Schicht.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Hebe die Begrenzungsrahmen auf, indem du den Versatz des Begrenzungsrahmens hinzufĂŒgst.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Hebe Punkte auf, indem du den Crop-Box-Versatz hinzufĂŒgst.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
Hebe Masken auf, indem du sie auf die ursprĂŒngliche BildgröĂe auffĂŒllst.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Entferne kleine unterbrochene Bereiche oder Löcher in einer Maske und gib die Maske und einen Ănderungsindikator zurĂŒck.
Quellcode in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Berechnet Boxen im XYXY-Format um Masken.
Bei einer leeren Maske wird [0,0,0,0] zurĂŒckgegeben. FĂŒr die Eingabeform C1xC2x...xHxW ist die Ausgabeform C1xC2x...x4.
Quellcode in ultralytics/models/sam/amg.py
Erstellt am 2023-11-12, Aktualisiert am 2024-05-08
Autoren: Burhan-Q (1), glenn-jocher (3), Laughing-q (1)