Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ def detect_and_estimate_objects(image):
|
|
| 46 |
logger.info("Image saved successfully for processing.")
|
| 47 |
|
| 48 |
# Step 1: YOLO detection
|
| 49 |
-
predictions = model.predict(temp_file_path, confidence=
|
| 50 |
class_count = {}
|
| 51 |
object_positions = []
|
| 52 |
|
|
|
|
| 46 |
logger.info("Image saved successfully for processing.")
|
| 47 |
|
| 48 |
# Step 1: YOLO detection
|
| 49 |
+
predictions = model.predict(temp_file_path, confidence=70, overlap=80).json()
|
| 50 |
class_count = {}
|
| 51 |
object_positions = []
|
| 52 |
|