slashtechno
/
wyzely-detect
mirror of https://github.com/slashtechno/wyzely-detect
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed `--detect-object` when specifying multiple objects

This commit is contained in:
slashtechno 2023-12-24 16:07:53 -06:00
parent 06bd1ccbd7
commit 82abe8b6d5
Signed by: slashtechno
GPG Key ID: 8EC1D9D9286C2B17
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
.vscode/launch.json vendored
View File

@ -14,6 +14,16 @@
], ],
"justMyCode": true "justMyCode": true
}, },
// {
// "name": "Quick, Specific Debug",
// "type": "python",
// "request": "launch",
// "module": "wyzely_detect",
// "args": [
// "--run-scale", "0.25", "--view-scale", "0.5", "--no-remove-representations", "--detect-object", "person", "--detect-object", "cell phone"
// ],
// "justMyCode": true
// },
{ {
// "name": "Python: Module", // "name": "Python: Module",
"name": "Full Debug", "name": "Full Debug",

2
wyzely_detect/utils/cli_args.py
View File

@ -154,7 +154,7 @@ def set_argparse():
object_detection = argparser.add_argument_group("Object Detection options") object_detection = argparser.add_argument_group("Object Detection options")
object_detection.add_argument( object_detection.add_argument(
"--detect-object", "--detect-object",
nargs="*", action="append",
default=[], default=[],
type=str, type=str,
help="The object(s) to detect. Must be something the model is trained to detect", help="The object(s) to detect. Must be something the model is trained to detect",