File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -56,7 +56,7 @@ def __init__(
5656 self .match_metric = match_metric
5757 self .intelligent_sorter = intelligent_sorter # enable sorting by area and confidence parameter
5858
59- # combinate detections of all patches
59+ # Combinate detections of all patches
6060 (
6161 self .detected_conf_list_full ,
6262 self .detected_xyxy_list_full ,
@@ -70,7 +70,7 @@ def __init__(
7070 ] # make str list
7171
7272 # Invoke the NMS for segmentation masks method for filtering predictions
73- if len (self .detected_masks_list_full )> 0 :
73+ if len (self .detected_masks_list_full ) > 0 :
7474
7575 self .filtered_indices = self .nms (
7676 self .detected_conf_list_full ,
@@ -96,11 +96,13 @@ def __init__(
9696 self .filtered_classes_id = [self .detected_cls_id_list_full [i ] for i in self .filtered_indices ]
9797 self .filtered_classes_names = [self .detected_cls_names_list_full [i ] for i in self .filtered_indices ]
9898
99+ # Masks filtering:
99100 if element_crops .segment and not element_crops .memory_optimize :
100101 self .filtered_masks = [self .detected_masks_list_full [i ] for i in self .filtered_indices ]
101102 else :
102103 self .filtered_masks = []
103104
105+ # Polygons filtering:
104106 if element_crops .segment and element_crops .memory_optimize :
105107 self .filtered_polygons = [self .detected_polygons_list_full [i ] for i in self .filtered_indices ]
106108 else :
Original file line number Diff line number Diff line change @@ -62,7 +62,7 @@ def __init__(
6262 classes_list = None ,
6363 segment = False ,
6464 shape_x = 700 ,
65- shape_y = 700 ,
65+ shape_y = 600 ,
6666 overlap_x = 25 ,
6767 overlap_y = 25 ,
6868 show_crops = False ,
You can’t perform that action at this time.
0 commit comments