Skip to content

Commit 3b61cf8

Browse files
committed
fix size to rect
1 parent 0b5188b commit 3b61cf8

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

patched_yolo_infer/nodes/CombineDetections.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff 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:

patched_yolo_infer/nodes/MakeCropsDetectThem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)