Riferimento per ultralytics/models/sam/amg.py
Nota
Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/amg .py. Se noti un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Restituisce un booleano tensor che indica se le caselle sono vicine al bordo del ritaglio.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Restituisce batch di dati dagli argomenti in ingresso.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Calcola il punteggio di stabilità per un gruppo di maschere.
Il punteggio di stabilità è l'IoU tra le maschere binarie ottenute dalla soglia dei loghi delle maschere previste ai valori alti e bassi. e bassi.
Note
- Una maschera è sempre contenuta all'interno dell'altra.
- Risparmia memoria evitando cast non necessari a torch.int64
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
Genera una griglia 2D di punti uniformemente distanziati nell'intervallo [0,1]x[0,1].
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Genera griglie di punti per tutti i livelli del raccolto.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Genera un elenco di caselle di ritaglio di diverse dimensioni.
Ogni strato ha (2i)2 caselle per lo strato iesimo.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Disincrosta le caselle di delimitazione aggiungendo l'offset della casella di ritaglio.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Disincrosta i punti aggiungendo l'offset della casella di ritaglio.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
Disincrosta le maschere facendole corrispondere alle dimensioni dell'immagine originale.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Rimuove piccole regioni scollegate o fori in una maschera, restituendo la maschera e un indicatore di modifica.
Codice sorgente in ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Calcola le caselle in formato XYXY intorno alle maschere.
Restituisce [0,0,0,0] per una maschera vuota. Per la forma di input C1xC2x...xHxW, la forma di output è C1xC2x...x4.