Spaces:
Running
Running
Made siglip-256 as default classifier
Browse files- app.py +2 -2
- dfine_jina_pipeline.py +1 -1
app.py
CHANGED
|
@@ -24,7 +24,7 @@ def run_dfine_classify(image, dfine_threshold, dfine_model_choice, classifier_ch
|
|
| 24 |
|
| 25 |
dfine_model = dfine_model_choice.strip().lower() if dfine_model_choice else "medium-obj2coco"
|
| 26 |
conf_thresh = float(siglip_threshold)
|
| 27 |
-
classifier = classifier_choice.strip() if classifier_choice else "siglip-
|
| 28 |
|
| 29 |
group_crops, known_crops, status = run_single_image(
|
| 30 |
image,
|
|
@@ -76,7 +76,7 @@ with gr.Blocks(title="Small Object Detection") as app:
|
|
| 76 |
|
| 77 |
classifier_dropdown = gr.Dropdown(
|
| 78 |
choices=["siglip-224", "siglip-256", "siglip-384"],
|
| 79 |
-
value="siglip-
|
| 80 |
label="Classifier model",
|
| 81 |
)
|
| 82 |
|
|
|
|
| 24 |
|
| 25 |
dfine_model = dfine_model_choice.strip().lower() if dfine_model_choice else "medium-obj2coco"
|
| 26 |
conf_thresh = float(siglip_threshold)
|
| 27 |
+
classifier = classifier_choice.strip() if classifier_choice else "siglip-256"
|
| 28 |
|
| 29 |
group_crops, known_crops, status = run_single_image(
|
| 30 |
image,
|
|
|
|
| 76 |
|
| 77 |
classifier_dropdown = gr.Dropdown(
|
| 78 |
choices=["siglip-224", "siglip-256", "siglip-384"],
|
| 79 |
+
value="siglip-256",
|
| 80 |
label="Classifier model",
|
| 81 |
)
|
| 82 |
|
dfine_jina_pipeline.py
CHANGED
|
@@ -568,7 +568,7 @@ def run_single_image(
|
|
| 568 |
crop_dedup_iou=0.35,
|
| 569 |
squarify=True,
|
| 570 |
min_display_conf=None,
|
| 571 |
-
classifier="siglip-
|
| 572 |
labels=None,
|
| 573 |
):
|
| 574 |
"""
|
|
|
|
| 568 |
crop_dedup_iou=0.35,
|
| 569 |
squarify=True,
|
| 570 |
min_display_conf=None,
|
| 571 |
+
classifier="siglip-256",
|
| 572 |
labels=None,
|
| 573 |
):
|
| 574 |
"""
|