Skip to content

Commit 7cb5714

Browse files
chore: Updated threshold from 501 to 100 on OD results upload (#723)
* Updated threshold from 501 to 100 on OD results upload * bump od examp test to large instance --------- Co-authored-by: Nan Wei <nan@kolena.io>
1 parent c45a166 commit 7cb5714

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

.circleci/continue_config.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -441,15 +441,15 @@ workflows:
441441
- example-test-dataset:
442442
matrix:
443443
parameters:
444-
subproject: [ text_summarization, image_retrieval_by_text ]
444+
subproject: [ text_summarization, image_retrieval_by_text, object_detection_2d ]
445445
resource-class: [ large ]
446446
python-version: [ "3.9.18" ]
447447
- example-test-dataset:
448448
context:
449449
- aws
450450
matrix:
451451
parameters:
452-
subproject: [ age_estimation, automatic_speech_recognition, classification, keypoint_detection, question_answering, rain_forecast, semantic_segmentation, speaker_diarization, object_detection_2d, semantic_textual_similarity, person_detection, crossing_pedestrian_detection, named_entity_recognition ]
452+
subproject: [ age_estimation, automatic_speech_recognition, classification, keypoint_detection, question_answering, rain_forecast, semantic_segmentation, speaker_diarization, semantic_textual_similarity, person_detection, crossing_pedestrian_detection, named_entity_recognition ]
453453
resource-class: [ small ]
454454
python-version: [ "3.9.18" ]
455455
- example-test-workflow:

kolena/_experimental/object_detection/dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,8 +276,8 @@ def _compute_metrics(
276276
)
277277
all_thresholds.extend(inf.score for inf in inferences)
278278

279-
if len(all_thresholds) >= 501:
280-
all_thresholds = list(np.linspace(min(all_thresholds), max(all_thresholds), 501))
279+
if len(all_thresholds) >= 100:
280+
all_thresholds = list(np.linspace(min(all_thresholds), max(all_thresholds), 100))
281281
else:
282282
all_thresholds = sorted(all_thresholds)
283283

0 commit comments

Comments
 (0)