muhammadsalmanalfaridzi commited on
Commit
3d78ecd
·
verified ·
1 Parent(s): 57c654b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- filtered_detections = detections.filter(strategy=sv.OverlapFilter.NON_MAX_MERGE, iou_threshold=0.5)
 
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)