Commit Graph
65 Commits
Author SHA1 Message Date
MediaPipe TeamandCopybara-Service c94de4032d Fix preprocess Callable typing
PiperOrigin-RevId: 515818356
2023-03-10 21:50:46 -08:00
MediaPipe TeamandCopybara-Service 2d8f937913 Improve docstring of image classifier model spec.
PiperOrigin-RevId: 515466722
2023-03-09 16:06:40 -08:00
Yuqi LiandCopybara-Service 9e7950a69a Make each model file in the model asset bundle to be aligned relative to the start of the file (alignment = 4).
PiperOrigin-RevId: 511624410
2023-02-22 16:15:24 -08:00
MediaPipe TeamandCopybara-Service 40b0dc960a Internal Changes
PiperOrigin-RevId: 511604893
2023-02-22 14:55:03 -08:00
MediaPipe TeamandCopybara-Service bdd1c24990 Refactor common methods into vision/core/image_utils.py and vision/core/test_utils.py
PiperOrigin-RevId: 509968910
2023-02-15 16:54:38 -08:00
MediaPipe TeamandCopybara-Service 99fc975f49 Add download model on demand to text classifier
PiperOrigin-RevId: 508441452
2023-02-09 12:15:17 -08:00
MediaPipe TeamandCopybara-Service 28f728bed5 Change gesture_recognizer to download model files on-demand from GCS.
PiperOrigin-RevId: 508424508
2023-02-09 11:13:06 -08:00
MediaPipe TeamandCopybara-Service d283e6a05a Support downloading model files on-demand from GCS in model_maker
PiperOrigin-RevId: 506174708
2023-01-31 18:43:35 -08:00
MediaPipe TeamandCopybara-Service afb0182935 Internal model maker change.
PiperOrigin-RevId: 504472342
2023-01-24 23:16:21 -08:00
MediaPipe TeamandCopybara-Service 4e135ccdb9 Internal Model Maker change.
PiperOrigin-RevId: 504315641
2023-01-24 10:38:25 -08:00
MediaPipe TeamandCopybara-Service 9cbb76939d Adds smaller MobileBERT model.
PiperOrigin-RevId: 501451414
2023-01-11 20:35:39 -08:00
MediaPipe TeamandCopybara-Service d40fa6b16d Internal Model Maker change.
PiperOrigin-RevId: 500758488
2023-01-09 11:04:45 -08:00
MediaPipe TeamandCopybara-Service e3131d7d78 Internal change
PiperOrigin-RevId: 499521620
2023-01-04 10:33:04 -08:00
MediaPipe TeamandCopybara-Service add5600d0d Internal change
PiperOrigin-RevId: 499351795
2023-01-03 17:20:46 -08:00
MediaPipe TeamandCopybara-Service 7e36a5e2ae Set filecmp.cmp(shallow=False) in model_maker unit tests.
PiperOrigin-RevId: 498218578
2022-12-28 11:25:28 -08:00
MediaPipe TeamandCopybara-Service 5a71b551e5 Remove duplicate and non-public api for model_maker
PiperOrigin-RevId: 497251246
2022-12-22 15:31:12 -08:00
MediaPipe TeamandCopybara-Service 36f054dfbe Internal model maker change
PiperOrigin-RevId: 497196512
2022-12-22 10:43:15 -08:00
MediaPipe TeamandCopybara-Service 5c0f548f5f Switches to tf.keras.optimizers.experimental.AdamW instead of the legacy AdamW.
PiperOrigin-RevId: 496821354
2022-12-20 20:53:13 -08:00
MediaPipe TeamandCopybara-Service a7b52d2c52 Internal changes
PiperOrigin-RevId: 496754449
2022-12-20 14:37:18 -08:00
MediaPipe TeamandCopybara-Service 3e6cd5d2bf Add support for customizing gesture recognizer layers
PiperOrigin-RevId: 496456160
2022-12-19 11:57:09 -08:00
MediaPipe TeamandCopybara-Service b4e1969e43 Add pip package builder for model_maker
PiperOrigin-RevId: 493989013
2022-12-08 14:03:15 -08:00
MediaPipe TeamandCopybara-Service a641ea12e1 Update gesture recognizer to new mediapipe tasks pipeline
PiperOrigin-RevId: 493950564
2022-12-08 11:32:42 -08:00
MediaPipe TeamandCopybara-Service 02aa162c9e Rename gesture_recognizer test_data to testdata to be consistent with rest of model_maker
PiperOrigin-RevId: 492246728
2022-12-01 11:13:19 -08:00
Scott ZhuandCopybara-Service a367753eda Internal change
PiperOrigin-RevId: 489824381
2022-11-20 10:41:20 -08:00
MediaPipe TeamandCopybara-Service bdf4078e89 Internal change
PiperOrigin-RevId: 489752009
2022-11-19 21:14:24 -08:00
MediaPipe TeamandCopybara-Service 977ee4272e Add public visibility to the model maker public API.
PiperOrigin-RevId: 489701768
2022-11-19 10:53:21 -08:00
MediaPipe TeamandCopybara-Service 524ac3ca61 Internal change for Model Maker
PiperOrigin-RevId: 489540387
2022-11-18 12:48:30 -08:00
MediaPipe TeamandCopybara-Service 1fb0902aa0 Update gesture_recognizer test
PiperOrigin-RevId: 489301508
2022-11-17 14:04:30 -08:00
MediaPipe TeamandCopybara-Service 74474d859e Update image_classifier demo with new ImageClassifierOption changes
PiperOrigin-RevId: 489031381
2022-11-16 14:00:39 -08:00
MediaPipe TeamandCopybara-Service cdd44e77b7 Internal change
PiperOrigin-RevId: 488969539
2022-11-16 10:07:40 -08:00
MediaPipe TeamandCopybara-Service 2f77bf44e3 Use train_data to evaluate accuracy of unit test for gesture_recognizer due to limited dataset size.
PiperOrigin-RevId: 488808942
2022-11-15 18:10:28 -08:00
MediaPipe TeamandCopybara-Service e65f21e2d8 Update the docstring to make it consistent with the model option update.
PiperOrigin-RevId: 488761331
2022-11-15 14:36:48 -08:00
MediaPipe TeamandCopybara-Service 1689112b23 Improve model_util_test code.
PiperOrigin-RevId: 488752497
2022-11-15 14:04:39 -08:00
MediaPipe TeamandCopybara-Service f14645cb06 Model maker gesture recognizer test changes
PiperOrigin-RevId: 488702055
2022-11-15 10:50:32 -08:00
MediaPipe TeamandCopybara-Service aafbf73c0a Renames model options in TextClassifier.
PiperOrigin-RevId: 488063797
2022-11-12 10:02:19 -08:00
MediaPipe TeamandCopybara-Service c2ac040a6c Adds a public import API for TextClassifier.
PiperOrigin-RevId: 487949023
2022-11-11 16:52:16 -08:00
MediaPipe TeamandCopybara-Service 3e05871f98 Open source Model Maker text tasks.
PiperOrigin-RevId: 487706929
2022-11-10 19:52:51 -08:00
MediaPipe TeamandCopybara-Service ec327cedcb Internal Changes
PiperOrigin-RevId: 487673720
2022-11-10 16:48:13 -08:00
MediaPipe TeamandCopybara-Service a5bcb97d88 Adds an AverageWordVecModel.
PiperOrigin-RevId: 487104909
2022-11-08 18:29:21 -08:00
MediaPipe TeamandCopybara-Service c31aaa94a6 Adds a BertClassifier.
PiperOrigin-RevId: 487086744
2022-11-08 16:49:46 -08:00
MediaPipe TeamandCopybara-Service 0917e8cb8e Support continual training image classifier from saved checkpoint files.
PiperOrigin-RevId: 487057612
2022-11-08 14:50:04 -08:00
MediaPipe TeamandCopybara-Service b3d19fa1af Use model bundle writer when exporting models in gesture recognizer
PiperOrigin-RevId: 487042776
2022-11-08 13:52:08 -08:00
MediaPipe TeamandCopybara-Service 4a6562d423 internal
PiperOrigin-RevId: 486853812
2022-11-07 22:39:23 -08:00
MediaPipe TeamandCopybara-Service 24d03451c7 Internal change - migration
PiperOrigin-RevId: 486853689
2022-11-07 22:36:22 -08:00
MediaPipe TeamandCopybara-Service c3bb4bb5da Refactor image classifier Create API by defining an ImageClassifierOption which wraps the model_spec, model_options, and hparams. Also migrate the definition of HParams by extending it from BaseHParams.
PiperOrigin-RevId: 486796657
2022-11-07 16:39:35 -08:00
MediaPipe TeamandCopybara-Service 6b54cae34c Internal change - cleanup
PiperOrigin-RevId: 486721059
2022-11-07 11:42:50 -08:00
MediaPipe TeamandCopybara-Service b2cc2cc60c Internal changes
PiperOrigin-RevId: 486014305
2022-11-03 17:23:20 -07:00
MediaPipe TeamandCopybara-Service d29c3d7512 Add metadata writer to image_classifier in model_maker
PiperOrigin-RevId: 485926985
2022-11-03 11:30:38 -07:00
MediaPipe TeamandCopybara-Service 0bf5110a7e Add a model_maker internal library directory.
PiperOrigin-RevId: 485665201
2022-11-02 12:31:58 -07:00
MediaPipe TeamandCopybara-Service 459214e6a3 Rename index_by_label to label_names.
PiperOrigin-RevId: 484956259
2022-10-30 22:26:17 -07:00