This commit is contained in:
corkborg
2023-12-28 03:05:00 +09:00
parent ebd8762e57
commit 878160d6b2
2 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -15,7 +15,7 @@ group.add_argument('--dir', help='ディレクトリ内の画像すべてに予
group.add_argument('--file', help='ファイルに対して予測を行う')
parser.add_argument(
'--threthold',
'--threshold',
type=float,
default=0.35,
help='予測値の足切り確率(デフォルトは0.35)')
@@ -40,7 +40,7 @@ def image_interrogate(image_path: Path):
"""
im = Image.open(image_path)
result = interrogator.interrogate(im)
return Interrogator.postprocess_tags(result[1], threshold=0.35)
return Interrogator.postprocess_tags(result[1], threshold=args.threshold)
if args.dir:
d = Path(args.dir)