81bbf66aab
* Add project code * Logger improvements * Improvements to web demo code * added create_wlasl_landmarks_dataset.py and xtract_mediapipe_landmarks.py * Fix rotation augmentation * fixed error in docstring, and removed unnecessary replace -1 -> 0 * Readme updates * Share base notebooks * Add notebooks and unify for different datasets * requirements update * fixes * Make evaluate more deterministic * Allow training with clearml * refactor preprocessing and apply linter * Minor fixes * Minor notebook tweaks * Readme updates * Fix PR comments * Remove unneeded code * Add banner to Readme --------- Co-authored-by: Gabriel Lema <gabriel.lema@xmartlabs.com>
22 lines
747 B
Python
22 lines
747 B
Python
from argparse import ArgumentParser
|
|
from preprocessing.create_wlasl_landmarks_dataset import parse_create_args, create
|
|
from preprocessing.extract_mediapipe_landmarks import parse_extract_args, extract
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main_parser = ArgumentParser()
|
|
subparser = main_parser.add_subparsers(dest="action")
|
|
create_subparser = subparser.add_parser("create")
|
|
extract_subparser = subparser.add_parser("extract")
|
|
parse_create_args(create_subparser)
|
|
parse_extract_args(extract_subparser)
|
|
|
|
args = main_parser.parse_args()
|
|
|
|
if args.action == "create":
|
|
create(args)
|
|
elif args.action == "extract":
|
|
extract(args)
|
|
else:
|
|
ValueError("action command must be either 'create' or 'extract'")
|