Skip to content

Commit cf2ef85

Browse files
committed
release v1.1.1
1 parent aacb7df commit cf2ef85

5 files changed

Lines changed: 13 additions & 10 deletions

File tree

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ element_crops = MakeCropsDetectThem(
8484
iou=0.7,
8585
resize_initial_size=True,
8686
)
87-
result = CombineDetections(element_crops, nms_threshold=0.05, match_metric='IOS')
87+
result = CombineDetections(element_crops, nms_threshold=0.25, match_metric='IOS')
8888

8989
# Final Results:
9090
img=result.image
@@ -122,7 +122,7 @@ Class implementing combining masks/boxes from multiple crops + NMS (Non-Maximum
122122
- **nms_threshold** (*float*): IoU/IoS threshold for non-maximum suppression.
123123
- **match_metric** (*str*): Matching metric, either 'IOU' or 'IOS'.
124124
- **intelligent_sorter** (*bool*): Enable sorting by area and rounded confidence parameter.
125-
If False, sorting will be done only by confidence (usual nms). (Dafault is False)
125+
If False, sorting will be done only by confidence (usual nms). (Dafault is True)
126126

127127

128128

examples/example_patch_based_inference.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@
9090
" element_crops (MakeCropsDetectThem): Object containing crop information.\n",
9191
" nms_threshold (float): IoU/IoS threshold for non-maximum suppression.\n",
9292
" match_metric (str): Matching metric, either 'IOU' or 'IOS'.\n",
93+
" intelligent_sorter (bool): Enable sorting by area and rounded confidence parameter. If False, sorting will be done only by confidence (usual nms). (Dafault is True)\n",
9394
"```"
9495
]
9596
},
@@ -1008,7 +1009,7 @@
10081009
"name": "python",
10091010
"nbconvert_exporter": "python",
10101011
"pygments_lexer": "ipython3",
1011-
"version": "3.10.13"
1012+
"version": "3.11.8"
10121013
}
10131014
},
10141015
"nbformat": 4,

patched_yolo_infer/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ element_crops = MakeCropsDetectThem(
7171
iou=0.7,
7272
resize_initial_size=True,
7373
)
74-
result = CombineDetections(element_crops, nms_threshold=0.05, match_metric='IOS')
74+
result = CombineDetections(element_crops, nms_threshold=0.25, match_metric='IOS')
7575

7676
# Final Results:
7777
img=result.image
@@ -108,7 +108,7 @@ Class implementing combining masks/boxes from multiple crops + NMS (Non-Maximum
108108
- **element_crops** (*MakeCropsDetectThem*): Object containing crop information.
109109
- **nms_threshold** (*float*): IoU/IoS threshold for non-maximum suppression.
110110
- **match_metric** (*str*): Matching metric, either 'IOU' or 'IOS'.
111-
- **intelligent_sorter** (*bool*): Enable sorting by area and rounded confidence parameter. If False, sorting will be done only by confidence (usual nms). (Dafault is False)
111+
- **intelligent_sorter** (*bool*): Enable sorting by area and rounded confidence parameter. If False, sorting will be done only by confidence (usual nms). (Dafault is True)
112112

113113
---
114114
### 2. Custom inference visualization:

patched_yolo_infer/nodes/CombineDetections.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ class CombineDetections:
1212
nms_threshold (float): IoU/IoS threshold for non-maximum suppression.
1313
match_metric (str): Matching metric, either 'IOU' or 'IOS'.
1414
intelligent_sorter (bool): Enable sorting by area and rounded confidence parameter.
15-
If False, sorting will be done only by confidence (usual nms). (Dafault False)
15+
If False, sorting will be done only by confidence (usual nms). (Dafault True)
1616
1717
Attributes:
1818
conf_treshold (float): Confidence threshold of yolov8.
@@ -40,7 +40,7 @@ def __init__(
4040
element_crops: MakeCropsDetectThem,
4141
nms_threshold=0.3,
4242
match_metric='IOS',
43-
intelligent_sorter=False
43+
intelligent_sorter=True
4444
) -> None:
4545
self.conf_treshold = element_crops.conf
4646
self.class_names = element_crops.class_names_dict

setup.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,13 @@
88
long_description = "\n" + fh.read()
99

1010

11-
VERSION = '1.1.0'
11+
VERSION = '1.1.1'
1212
DESCRIPTION = '''YOLO-Patch-Based-Inference for detection/segmentation of small objects in images.'''
1313

1414
setup(
1515
name="patched_yolo_infer",
1616
version=VERSION,
17-
license="MIT",
17+
license="AGPL-3.0 license",
1818
url="https://github.com/Koldim2001/YOLO-Patch-Based-Inference",
1919
author="Koldim2001",
2020
description=DESCRIPTION,
@@ -46,12 +46,14 @@
4646
"slicing inference",
4747
"inference visualization",
4848
"patchify",
49+
"ultralytics",
50+
"SAHI",
4951
],
5052
classifiers=[
5153
"Development Status :: 5 - Production/Stable",
5254
"Intended Audience :: Developers",
5355
"Programming Language :: Python :: 3",
54-
"License :: OSI Approved :: MIT License",
56+
"License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+) (AGPL-3.0)",
5557
"Operating System :: OS Independent",
5658
],
5759
)

0 commit comments

Comments
 (0)