Update app.py
Browse files
app.py
CHANGED
@@ -28,13 +28,14 @@ def detect_objects(image):
|
|
28 |
return sv.Detections.from_inference(predictions)
|
29 |
|
30 |
# Menggunakan InferenceSlicer
|
31 |
-
slicer = sv.InferenceSlicer(callback=callback)
|
32 |
|
33 |
# Proses gambar dengan slicer
|
34 |
detections = slicer(img)
|
35 |
|
36 |
# Filter deteksi yang tumpang tindih (gunakan NMM atau tanpa filter)
|
37 |
-
|
|
|
38 |
|
39 |
# Annotasi gambar dengan deteksi
|
40 |
annotated_image = sv.BoxAnnotator().annotate(scene=img.copy(), detections=filtered_detections)
|
|
|
28 |
return sv.Detections.from_inference(predictions)
|
29 |
|
30 |
# Menggunakan InferenceSlicer
|
31 |
+
slicer = sv.InferenceSlicer(callback=callback, overlap_wh=(0.2, 0.2)) # Replacing overlap_ratio_wh
|
32 |
|
33 |
# Proses gambar dengan slicer
|
34 |
detections = slicer(img)
|
35 |
|
36 |
# Filter deteksi yang tumpang tindih (gunakan NMM atau tanpa filter)
|
37 |
+
overlap_filter = sv.OverlapFilter(strategy=sv.OverlapFilter.NON_MAX_MERGE, iou_threshold=0.5)
|
38 |
+
filtered_detections = overlap_filter.filter(detections)
|
39 |
|
40 |
# Annotasi gambar dengan deteksi
|
41 |
annotated_image = sv.BoxAnnotator().annotate(scene=img.copy(), detections=filtered_detections)
|