Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ def process_image(image, show_boxes=True):
|
|
48 |
image = np.array(image)
|
49 |
|
50 |
# Run inference with specific parameters
|
51 |
-
results = model.predict(image, imgsz=320, conf=0.
|
52 |
|
53 |
# Lists to store center points of knots
|
54 |
centers_x = []
|
|
|
48 |
image = np.array(image)
|
49 |
|
50 |
# Run inference with specific parameters
|
51 |
+
results = model.predict(image, imgsz=320, conf=0.4, iou=0.9)[0]
|
52 |
|
53 |
# Lists to store center points of knots
|
54 |
centers_x = []
|