Minor fixes (#162)

This commit is contained in:
Laughing
2023-01-09 20:41:34 +08:00
committed by GitHub
parent 9a2f67b3b4
commit 20fee4100c
3 changed files with 15 additions and 53 deletions

View File

@ -118,6 +118,7 @@ class YOLO:
**kwargs : Any other args accepted by the predictors. To see all args check 'configuration' section in docs
"""
overrides = self.overrides.copy()
overrides["conf"] = 0.25
overrides.update(kwargs)
overrides["mode"] = "predict"
predictor = self.PredictorClass(overrides=overrides)