diff --git a/facetracker.py b/facetracker.py index 83b0088..19c8dee 100644 --- a/facetracker.py +++ b/facetracker.py @@ -39,7 +39,7 @@ parser.add_argument("--video-fps", type=float, help="This sets the frame rate of parser.add_argument("--raw-rgb", type=int, help="When this is set, raw RGB frames of the size given with \"-W\" and \"-H\" are read from standard input instead of reading a video", default=0) parser.add_argument("--log-data", help="You can set a filename to which tracking data will be logged here", default="") parser.add_argument("--log-output", help="You can set a filename to console output will be logged here", default="") -parser.add_argument("--model", type=int, help="This can be used to select the tracking model. Higher numbers are models with better tracking quality, but slower speed. Models 1 and 0 tend to be too rigid for expression and blink detection. Model -2 is roughly equivalent to model 1, but faster. Model -3 is between models 0 and -1.", default=3, choices=[-3, -2, -1, 0, 1, 2, 3]) +parser.add_argument("--model", type=int, help="This can be used to select the tracking model. Higher numbers are models with better tracking quality, but slower speed, except for model 4, which is wink optimized. Models 1 and 0 tend to be too rigid for expression and blink detection. Model -2 is roughly equivalent to model 1, but faster. Model -3 is between models 0 and -1.", default=3, choices=[-3, -2, -1, 0, 1, 2, 3, 4]) parser.add_argument("--model-dir", help="This can be used to specify the path to the directory containing the .onnx model files", default=None) parser.add_argument("--gaze-tracking", type=int, help="When set to 1, experimental blink detection and gaze tracking are enabled, which makes things slightly slower", default=1) parser.add_argument("--face-id-offset", type=int, help="When set, this offset is added to all face ids, which can be useful for mixing tracking data from multiple network sources", default=0) diff --git a/models/lm_model4_opt.onnx b/models/lm_model4_opt.onnx new file mode 100644 index 0000000..cc92fc0 Binary files /dev/null and b/models/lm_model4_opt.onnx differ diff --git a/tracker.py b/tracker.py index 861d684..e41681a 100644 --- a/tracker.py +++ b/tracker.py @@ -506,7 +506,8 @@ class Tracker(): "lm_model0_opt.onnx", "lm_model1_opt.onnx", "lm_model2_opt.onnx", - "lm_model3_opt.onnx" + "lm_model3_opt.onnx", + "lm_model4_opt.onnx" ] model = "lm_modelT_opt.onnx" if model_type >= 0: