Tài liệu tham khảo cho ultralytics/models/sam/amg.py
Ghi
Tệp này có sẵn tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/Mô hình/sam/amg.py. Nếu bạn phát hiện ra một vấn đề, vui lòng giúp khắc phục nó bằng cách đóng góp Yêu cầu 🛠️ kéo. Cảm ơn bạn 🙏 !
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
Trả về boolean tensor cho biết nếu các hộp ở gần cạnh cắt.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
Cho ra các lô dữ liệu từ các đối số đầu vào.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
Tính điểm ổn định cho một lô khẩu trang.
Điểm ổn định là IoU giữa các mặt nạ nhị phân thu được bằng cách ngưỡng nhật ký mặt nạ dự đoán ở mức cao và giá trị thấp.
Ghi chú
- Một mặt nạ luôn được chứa bên trong mặt nạ kia.
- Tiết kiệm bộ nhớ bằng cách ngăn truyền không cần thiết để torch.int64
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
Tạo lưới 2D gồm các điểm cách đều nhau trong phạm vi [0,1]x[0,1].
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
Tạo lưới điểm cho tất cả các lớp cắt.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
Tạo danh sách các hộp cắt có kích thước khác nhau.
Mỗi layer có (2i)2 box cho layer thứ i.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
Bỏ cắt các hộp giới hạn bằng cách thêm độ lệch hộp cắt.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
Bỏ xén điểm bằng cách thêm độ lệch của hộp cắt.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
Bỏ cắt mặt nạ bằng cách đệm chúng về kích thước hình ảnh gốc.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
Loại bỏ các vùng hoặc lỗ nhỏ bị ngắt kết nối trên mặt nạ, trả lại mặt nạ và chỉ báo sửa đổi.
Mã nguồn trong ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
Tính toán các hộp ở định dạng XYXY xung quanh mặt nạ.
Trả lại [0,0,0,0] cho mặt nạ trống. Đối với hình dạng đầu vào C1xC2x ... xHxW, hình dạng đầu ra là C1xC2x ... x4.