Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ def detect_objects(img: Image.Image):
|
|
12 |
|
13 |
target_sizes = torch.tensor([img.size[::-1]]) #
|
14 |
results = processor.post_process_object_detection(
|
15 |
-
outputs, target_sizes=target_sizes, threshold=0.
|
16 |
)[0]
|
17 |
|
18 |
draw = ImageDraw.Draw(img)
|
|
|
12 |
|
13 |
target_sizes = torch.tensor([img.size[::-1]]) #
|
14 |
results = processor.post_process_object_detection(
|
15 |
+
outputs, target_sizes=target_sizes, threshold=0.3
|
16 |
)[0]
|
17 |
|
18 |
draw = ImageDraw.Draw(img)
|