Referencia para ultralytics/models/sam/amg.py
Nota
Este archivo está disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/amg .py. Si detectas algún problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 🛠️. ¡Gracias 🙏!
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Devuelve un booleano tensor que indica si las cajas están cerca del borde de recorte.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Produce lotes de datos a partir de los argumentos de entrada.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Calcula la puntuación de estabilidad de un lote de máscaras.
La puntuación de estabilidad es el IoU entre las máscaras binarias obtenidas mediante el umbral de los logits de máscara predichos en valores altos y bajos.
Notas
- Una máscara siempre está contenida dentro de la otra.
- Ahorra memoria evitando la conversión innecesaria a torch.int64
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
Genera una malla 2D de puntos espaciados uniformemente en el intervalo [0,1]x[0,1].
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Genera rejillas de puntos para todas las capas de cultivo.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Genera una lista de cajas de recorte de diferentes tamaños.
Cada capa tiene (2i)2 casillas para la capa i-ésima.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Desencuadra los cuadros delimitadores añadiendo el desplazamiento del cuadro de recorte.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Desencuadra los puntos añadiendo el desplazamiento del cuadro de recorte.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
Recorta las máscaras rellenándolas al tamaño original de la imagen.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Elimina pequeñas regiones desconectadas o agujeros en una máscara, devolviendo la máscara y un indicador de modificación.
Código fuente en ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Calcula recuadros en formato XYXY alrededor de las máscaras.
Devuelve [0,0,0,0] para una máscara vacía. Para la forma de entrada C1xC2x...xHxW, la forma de salida es C1xC2x...x4.