рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/models/sam/amg.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/рдореЙрдбрд▓/sam/amg.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.models.sam.amg.is_box_near_crop_edge(boxes, crop_box, orig_box, atol=20.0)
рдПрдХ рдмреВрд▓рд┐рдпрди рд▓реМрдЯрд╛рдПрдВ tensor рдпрд╣ рджрд░реНрд╢рд╛рддрд╛ рд╣реИ рдХрд┐ рдмрдХреНрд╕реЗ рдлрд╕рд▓ рдХреЗ рдХрд┐рдирд╛рд░реЗ рдХреЗ рдкрд╛рд╕ рд╣реИрдВ рдпрд╛ рдирд╣реАрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batch_iterator(batch_size, *args)
рдЗрдирдкреБрдЯ рддрд░реНрдХреЛрдВ рд╕реЗ рдбреЗрдЯрд╛ рдХреЗ рдмреИрдЪ рдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.calculate_stability_score(masks, mask_threshold, threshold_offset)
рдорд╛рд╕реНрдХ рдХреЗ рдПрдХ рдмреИрдЪ рдХреЗ рд▓рд┐рдП рд╕реНрдерд┐рд░рддрд╛ рд╕реНрдХреЛрд░ рдХреА рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИред
рд╕реНрдерд┐рд░рддрд╛ рд╕реНрдХреЛрд░ рдЙрдЪреНрдЪ рдкрд░ рдЕрдиреБрдорд╛рдирд┐рдд рдореБрдЦреМрдЯрд╛ рд▓реЙрдЧреНрд╕ рдХреЛ рдереНрд░реЗрд╕рд╣реЛрд▓реНрдб рдХрд░рдХреЗ рдкреНрд░рд╛рдкреНрдд рдмрд╛рдЗрдирд░реА рдорд╛рд╕реНрдХ рдХреЗ рдмреАрдЪ IoU рд╣реИ рдФрд░ рдХрдо рдореВрд▓реНрдпред
рдиреЛрдЯреНрд╕
- рдПрдХ рдореБрдЦреМрдЯрд╛ рд╣рдореЗрд╢рд╛ рджреВрд╕рд░реЗ рдХреЗ рдЕрдВрджрд░ рдирд┐рд╣рд┐рдд рд╣реЛрддрд╛ рд╣реИред
- рдЕрдирд╛рд╡рд╢реНрдпрдХ рдХрд╛рд╕реНрдЯ рдХреЛ рд░реЛрдХрдХрд░ рдореЗрдореЛрд░реА рдХреЛ рдмрдЪрд╛рдПрдВ torch.int64
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_point_grid(n_per_side)
[2]x[0,1] рд╢реНрд░реЗрдгреА рдореЗрдВ рд╕рдорд╛рди рд░реВрдк рд╕реЗ рджреВрд░реА рд╡рд╛рд▓реЗ рдмрд┐рдВрджреБрдУрдВ рдХрд╛ 0,1D рдЧреНрд░рд┐рдб рдЙрддреНрдкрдиреНрди рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.build_all_layer_point_grids(n_per_side, n_layers, scale_per_layer)
рд╕рднреА рдлрд╕рд▓ рдкрд░рддреЛрдВ рдХреЗ рд▓рд┐рдП рдмрд┐рдВрджреБ рдЧреНрд░рд┐рдб рдЙрддреНрдкрдиреНрди рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.generate_crop_boxes(im_size, n_layers, overlap_ratio)
рд╡рд┐рднрд┐рдиреНрди рдЖрдХрд╛рд░реЛрдВ рдХреЗ рдлрд╕рд▓ рдмрдХреНрд╕реЗ рдХреА рдПрдХ рд╕реВрдЪреА рдЙрддреНрдкрдиреНрди рдХрд░рддрд╛ рд╣реИред
рдкреНрд░рддреНрдпреЗрдХ рдкрд░рдд рдореЗрдВ ith рдкрд░рдд рдХреЗ рд▓рд┐рдП (2i)2 рдмреЙрдХреНрд╕ рд╣реЛрддреЗ рд╣реИрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_boxes_xyxy(boxes, crop_box)
рдХреНрд░реЙрдк рдмреЙрдХреНрд╕ рдСрдлрд╝рд╕реЗрдЯ рдЬреЛрдбрд╝рдХрд░ рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕ рдХреЛ рдЕрдирдХреНрд░реЙрдк рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_points(points, crop_box)
рдХреНрд░реЙрдк рдмреЙрдХреНрд╕ рдСрдлрд╝рд╕реЗрдЯ рдЬреЛрдбрд╝рдХрд░ рдЕрдВрдХ рдЕрдирдХреНрд░реЙрдк рдХрд░реЗрдВ.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.uncrop_masks(masks, crop_box, orig_h, orig_w)
рдорд╛рд╕реНрдХ рдХреЛ рдореВрд▓ рдЫрд╡рд┐ рдЖрдХрд╛рд░ рдореЗрдВ рдкреИрдбрд┐рдВрдЧ рдХрд░рдХреЗ рдЕрдирдХреНрд░реЙрдк рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.remove_small_regions(mask, area_thresh, mode)
рдПрдХ рдореБрдЦреМрдЯрд╛ рдореЗрдВ рдЫреЛрдЯреЗ рдбрд┐рд╕реНрдХрдиреЗрдХреНрдЯ рдХрд┐рдП рдЧрдП рдХреНрд╖реЗрддреНрд░реЛрдВ рдпрд╛ рдЫреЗрджреЛрдВ рдХреЛ рд╣рдЯрд╛ рджреЗрдВ, рдореБрдЦреМрдЯрд╛ рдФрд░ рдПрдХ рд╕рдВрд╢реЛрдзрди рд╕рдВрдХреЗрддрдХ рд▓реМрдЯрд╛рдПрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
ultralytics.models.sam.amg.batched_mask_to_box(masks)
рдорд╛рд╕реНрдХ рдХреЗ рдЪрд╛рд░реЛрдВ рдУрд░ XYXY рдкреНрд░рд╛рд░реВрдк рдореЗрдВ рдмрдХреНрд╕реЗ рдХреА рдЧрдгрдирд╛ рдХрд░рддрд╛ рд╣реИред
рдЦрд╛рд▓реА рдорд╛рд╕реНрдХ рдХреЗ рд▓рд┐рдП [0,0,0,0] рд▓реМрдЯрд╛рдПрдВред рдЗрдирдкреБрдЯ рдЖрдХрд╛рд░ C1xC2x рдХреЗ рд▓рд┐рдП... xHxW, рдЖрдЙрдЯрдкреБрдЯ рдЖрдХрд╛рд░ C1xC2x рд╣реИ... рдПрдХреНрд╕4.
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/models/sam/amg.py
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (3), рд▓рд╛рдлрд┐рдВрдЧ-рдХреНрдпреВ (1)