Skip to content

Commit

Permalink
Update CHANGELOG.md and docs
Browse files Browse the repository at this point in the history
  • Loading branch information
jaegukhyun committed Jul 24, 2023
1 parent 1039ce4 commit 65d7866
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 1 deletion.
2 changes: 1 addition & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ All notable changes to this project will be documented in this file.

### New features

-
- Enable configurable confidence threshold for otx eval and export(<https://github.com/openvinotoolkit/training_extensions/pull/2388>)

### Enhancements

Expand Down
16 changes: 16 additions & 0 deletions docs/source/guide/tutorials/base/how_to_train/detection.rst
Original file line number Diff line number Diff line change
Expand Up @@ -359,6 +359,22 @@ using ``otx eval`` and passing the IR model path to the ``--load-weights`` param
Performance(score: 0.5487693710118504, dashboard: (1 metric groups))
4. ``Optional`` Additionally, we can tune confidence threshold via the command line.
Learn more about template-specific parameters using ``otx export params --help``.

For example, if there are too many False-Positive predictions (there we have a prediction, but don't have annotated object for it), we can suppress its number by increasing the confidence threshold as it is shown below.

Please note, by default, the optimal confidence threshold is detected based on validation results to maximize the final F1 metric. To set a custom confidence threshold, please disable ``result_based_confidence_threshold`` option.

.. code-block::
(otx) ...$ otx export --load-weights ../outputs/weights.pth \
--output ../outputs
params \
--postprocessing.confidence_threshold 0.5 \
--postprocessing.result_based_confidence_threshold false
*************
Optimization
*************
Expand Down

0 comments on commit 65d7866

Please sign in to comment.