Skip to content
Snippets Groups Projects

Implement temperature scaling on dan

Merged Thibault Lavigne requested to merge dan-temperature-scaling into main
3 files
+ 63
7
Compare changes
  • Side-by-side
  • Inline
Files
3
+ 14
0
@@ -55,6 +55,20 @@ def add_predict_parser(subcommands) -> None:
required=False,
help="Image scaling factor before feeding it to DAN",
)
parser.add_argument(
"--image-max-width",
type=int,
default=1800,
required=False,
help="Image resizing before feeding it to DAN",
)
parser.add_argument(
"--temperature",
type=float,
default=1.0,
help="Temperature scaling scalar parameter",
required=True,
)
parser.add_argument(
"--confidence-score",
action="store_true",
Loading