Commit ·
9a44e5f
1
Parent(s): c7d0d4b
fix
Browse files
app/routers/__pycache__/segment.cpython-311.pyc
CHANGED
|
Binary files a/app/routers/__pycache__/segment.cpython-311.pyc and b/app/routers/__pycache__/segment.cpython-311.pyc differ
|
|
|
app/routers/embedd.py
CHANGED
|
@@ -108,12 +108,12 @@ async def upload_image_binary(request: Request, use_segmentation: bool = True):
|
|
| 108 |
im0, img = detector.preprocess_image_binary(image)
|
| 109 |
detections = detector.infer(img)
|
| 110 |
segmented_image = detector.process_detections(im0, detections)
|
| 111 |
-
print("with segmentation", segmented_image)
|
| 112 |
|
| 113 |
-
|
| 114 |
if segmented_image is None:
|
|
|
|
| 115 |
embedding = embedding_model(image)
|
| 116 |
else:
|
|
|
|
| 117 |
embedding = embedding_model(segmented_image)
|
| 118 |
|
| 119 |
return {
|
|
|
|
| 108 |
im0, img = detector.preprocess_image_binary(image)
|
| 109 |
detections = detector.infer(img)
|
| 110 |
segmented_image = detector.process_detections(im0, detections)
|
|
|
|
| 111 |
|
|
|
|
| 112 |
if segmented_image is None:
|
| 113 |
+
print("No segmentation")
|
| 114 |
embedding = embedding_model(image)
|
| 115 |
else:
|
| 116 |
+
print("with segmentation")
|
| 117 |
embedding = embedding_model(segmented_image)
|
| 118 |
|
| 119 |
return {
|