Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -146,7 +146,7 @@ def yolo_detect(image: Union[str, Path, int, Image.Image, list, tuple, np.ndarra
|
|
146 |
|
147 |
def detect_reference_square(img: np.ndarray):
|
148 |
t = time.time()
|
149 |
-
res = reference_detector_global.predict(img, conf=0.
|
150 |
if not res or len(res) == 0 or len(res[0].boxes) == 0:
|
151 |
raise ReferenceBoxNotDetectedError("Reference box not detected in the image.")
|
152 |
print("Reference detection completed in {:.2f} seconds".format(time.time() - t))
|
|
|
146 |
|
147 |
def detect_reference_square(img: np.ndarray):
|
148 |
t = time.time()
|
149 |
+
res = reference_detector_global.predict(img, conf=0.30)
|
150 |
if not res or len(res) == 0 or len(res[0].boxes) == 0:
|
151 |
raise ReferenceBoxNotDetectedError("Reference box not detected in the image.")
|
152 |
print("Reference detection completed in {:.2f} seconds".format(time.time() - t))
|