Default confidences predict 0.25, val 0.001 (#149)

This commit is contained in:
Glenn Jocher 2023-01-05 15:56:52 +01:00 committed by GitHub
parent 3cbf3ec455
commit 314da263c7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 2 deletions

View file

@ -77,7 +77,8 @@ class BasePredictor:
name = self.args.name or f"{self.args.mode}"
self.save_dir = increment_path(Path(project) / name, exist_ok=self.args.exist_ok)
(self.save_dir / 'labels' if self.args.save_txt else self.save_dir).mkdir(parents=True, exist_ok=True)
if self.args.conf is None:
self.args.conf = 0.25 # default conf=0.25
self.done_setup = False
# Usable if setup is done