Initial codebase (#1)
* 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>
This commit is contained in:
21
preprocessing.py
Normal file
21
preprocessing.py
Normal file
@@ -0,0 +1,21 @@
|
||||
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'")
|
||||
Reference in New Issue
Block a user