Skip to content

Commit

Permalink
Added MoveNet as an external model reference (#1141)
Browse files Browse the repository at this point in the history
* add center padding to size matcher

* add test for center padding

* add ensure_float option to inference layer

* reformat resizing and test_resizing

* add MoveNet as an external model inference

* add the movenet to the from_model_paths

* add tests

* add comments to movenet predictor

* add tensorflow_hub to the requirements.txt

* modified default video path

* resolved most of the comments except expanding the predictor

* expanded Predictor.from_model_paths function to include any pre-trained models.

* add test_load_model

* added from_trained_models in class Predictor and modified test_load_model for it.

* modified test_load_model to be more generalized.

* moved pretrained model from Predictor.from_trained_model to Predictor.from_model_paths and added a test for it.

* Fix Predictor.from_model_paths and tests

* Rename load_movenet_model to make_model_movenet

* minor clean-up

* Remove redundant operation

* Replace existing constants with fixtures

* Handle loading movenet models via load_model API

* Clean-up doc strings

---------

Co-authored-by: Liezl Maree <[email protected]>
  • Loading branch information
jiayinghsu and roomrys authored Jan 31, 2023
1 parent e204dae commit 833c2d5
Show file tree
Hide file tree
Showing 6 changed files with 414 additions and 22 deletions.
1 change: 1 addition & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ cattrs==1.1.1
jsonpickle==1.2
networkx
tensorflow>=2.6.3,<2.9.0; platform_machine != 'arm64'
tensorflow_hub
tensorflow-macos==2.9.2; sys_platform == 'darwin' and platform_machine == 'arm64'
tensorflow-metal==0.5.0; sys_platform == 'darwin' and platform_machine == 'arm64'
h5py>=3.1.0,<=3.7.0
Expand Down
Loading

0 comments on commit 833c2d5

Please sign in to comment.