orik-ss commited on
Commit
d5c03fb
·
1 Parent(s): 33708c6

Made siglip-256 as default classifier

Browse files
Files changed (2) hide show
  1. app.py +2 -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-224"
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-224",
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-224",
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
  """