Referência para ultralytics/models/sam/amg.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ sam/amg .py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Devolve um booleano tensor que indica se as caixas estão perto da margem de corte.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Produz lotes de dados a partir dos argumentos de entrada.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Calcula a pontuação de estabilidade para um lote de máscaras.
A pontuação de estabilidade é o IoU entre as máscaras binárias obtidas por limiarização dos logits da máscara prevista em valores altos e baixos. e baixos.
Notas
- Uma máscara está sempre contida dentro da outra.
- Poupa memória evitando a conversão desnecessária para torch.int64
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
Gera uma grelha 2D de pontos uniformemente espaçados no intervalo [0,1]x[0,1].
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Gera grelhas de pontos para todas as camadas de cultura.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Gera uma lista de caixas de corte de diferentes tamanhos.
Cada camada tem (2i)2 casas para a i-ésima camada.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Desfaz o corte das caixas delimitadoras adicionando o offset da caixa de corte.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Desfaz o corte de pontos adicionando o offset da caixa de corte.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
Desdobra as máscaras, preenchendo-as com o tamanho da imagem original.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Remove pequenas regiões desconectadas ou buracos numa máscara, devolvendo a máscara e um indicador de modificação.
Código fonte em ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Calcula caixas no formato XYXY em torno de máscaras.
Devolve [0,0,0,0] para uma máscara vazia. Para a forma de entrada C1xC2x...xHxW, a forma de saída é C1xC2x...x4.
Código fonte em ultralytics/models/sam/amg.py
Criado em 2023-11-12, Atualizado em 2024-05-08
Autores: Burhan-Q (1), glenn-jocher (3), Laughing-q (1)