Référence pour ultralytics/models/sam/amg.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/amg .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Renvoie un booléen tensor indiquant si les boîtes sont proches du bord de la culture.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Produit des lots de données à partir des arguments d'entrée.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Calcule le score de stabilité pour un lot de masques.
Le score de stabilité est l'IoU entre les masques binaires obtenus par seuillage des logits des masques prédits à des valeurs élevées et basses. et faibles.
Notes
- Un masque est toujours contenu dans l'autre.
- Économise de la mémoire en évitant les castings inutiles vers torch.int64
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
Génère une grille 2D de points régulièrement espacés dans l'intervalle [0,1]x[0,1].
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Génère des grilles de points pour toutes les couches de cultures.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Génère une liste de boîtes de culture de différentes tailles.
Chaque couche comporte (2i)2 cases pour la ième couche.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Découpe les boîtes de délimitation en ajoutant le décalage de la boîte de récolte.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Découpe les points en ajoutant le décalage de la boîte de culture.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
DĂ©coupe les masques en les ramenant Ă la taille de l'image originale.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Enlève les petites régions déconnectées ou les trous dans un masque, en renvoyant le masque et un indicateur de modification.
Code source dans ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Calcule des boîtes au format XYXY autour des masques.
Retourne [0,0,0,0] pour un masque vide. Pour la forme d'entrée C1xC2x...xHxW, la forme de sortie est C1xC2x...x4.
Code source dans ultralytics/models/sam/amg.py
Créé le 2023-11-12, Mis à jour le 2024-05-08
Auteurs : Burhan-Q (1), glenn-jocher (3), Laughing-q (1)