Commit Graph

88 Commits

Author SHA1 Message Date
MediaPipe Team
1fb0902aa0 Update gesture_recognizer test
PiperOrigin-RevId: 489301508
2022-11-17 14:04:30 -08:00
MediaPipe Team
74474d859e Update image_classifier demo with new ImageClassifierOption changes
PiperOrigin-RevId: 489031381
2022-11-16 14:00:39 -08:00
MediaPipe Team
cdd44e77b7 Internal change
PiperOrigin-RevId: 488969539
2022-11-16 10:07:40 -08:00
MediaPipe Team
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 Team
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 Team
1689112b23 Improve model_util_test code.
PiperOrigin-RevId: 488752497
2022-11-15 14:04:39 -08:00
MediaPipe Team
f14645cb06 Model maker gesture recognizer test changes
PiperOrigin-RevId: 488702055
2022-11-15 10:50:32 -08:00
MediaPipe Team
aafbf73c0a Renames model options in TextClassifier.
PiperOrigin-RevId: 488063797
2022-11-12 10:02:19 -08:00
MediaPipe Team
c2ac040a6c Adds a public import API for TextClassifier.
PiperOrigin-RevId: 487949023
2022-11-11 16:52:16 -08:00
MediaPipe Team
3e05871f98 Open source Model Maker text tasks.
PiperOrigin-RevId: 487706929
2022-11-10 19:52:51 -08:00
MediaPipe Team
ec327cedcb Internal Changes
PiperOrigin-RevId: 487673720
2022-11-10 16:48:13 -08:00
MediaPipe Team
a5bcb97d88 Adds an AverageWordVecModel.
PiperOrigin-RevId: 487104909
2022-11-08 18:29:21 -08:00
MediaPipe Team
c31aaa94a6 Adds a BertClassifier.
PiperOrigin-RevId: 487086744
2022-11-08 16:49:46 -08:00
MediaPipe Team
0917e8cb8e Support continual training image classifier from saved checkpoint files.
PiperOrigin-RevId: 487057612
2022-11-08 14:50:04 -08:00
MediaPipe Team
b3d19fa1af Use model bundle writer when exporting models in gesture recognizer
PiperOrigin-RevId: 487042776
2022-11-08 13:52:08 -08:00
MediaPipe Team
4a6562d423 internal
PiperOrigin-RevId: 486853812
2022-11-07 22:39:23 -08:00
MediaPipe Team
24d03451c7 Internal change - migration
PiperOrigin-RevId: 486853689
2022-11-07 22:36:22 -08:00
MediaPipe Team
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 Team
6b54cae34c Internal change - cleanup
PiperOrigin-RevId: 486721059
2022-11-07 11:42:50 -08:00
MediaPipe Team
b2cc2cc60c Internal changes
PiperOrigin-RevId: 486014305
2022-11-03 17:23:20 -07:00
MediaPipe Team
d29c3d7512 Add metadata writer to image_classifier in model_maker
PiperOrigin-RevId: 485926985
2022-11-03 11:30:38 -07:00
MediaPipe Team
0bf5110a7e Add a model_maker internal library directory.
PiperOrigin-RevId: 485665201
2022-11-02 12:31:58 -07:00
MediaPipe Team
459214e6a3 Rename index_by_label to label_names.
PiperOrigin-RevId: 484956259
2022-10-30 22:26:17 -07:00
MediaPipe Team
a28c9d2c26 Set steps_per_epoch to None when calling model.fit() method for image classifier.
PiperOrigin-RevId: 483764377
2022-10-25 13:44:03 -07:00
MediaPipe Team
ec2a34d2a4 Replace pytype_struct_contrib_test by py_strict_test. Also remove unnecessary BUILD attributes.
PiperOrigin-RevId: 483237371
2022-10-23 16:45:02 -07:00
MediaPipe Team
e453c3b206 Migrates image_preprocessing from core to vision.
PiperOrigin-RevId: 482856336
2022-10-21 12:16:54 -07:00
MediaPipe Team
55ba23ce9a Cleans up unused function from image_classifier's Dataset.
PiperOrigin-RevId: 482703775
2022-10-20 23:59:20 -07:00
MediaPipe Team
348c4e6652 Add a test to test saving and loading the gesture recognizer model.
Also update the loss function init method to resolve a bug when loading model.

PiperOrigin-RevId: 482702020
2022-10-20 23:45:26 -07:00
MediaPipe Team
086fc442fd Internal change
PiperOrigin-RevId: 482697999
2022-10-20 23:15:17 -07:00
MediaPipe Team
e71638cf67 Add an export_tflite API to gesture recognizer model maker library.
PiperOrigin-RevId: 482527017
2022-10-20 10:21:34 -07:00
MediaPipe Team
3d588bae8b Internal change
PiperOrigin-RevId: 482259130
2022-10-19 11:34:32 -07:00
MediaPipe Team
f2821d840d Improve image classifier model maker documentation and replace the legacy code.
PiperOrigin-RevId: 481979922
2022-10-18 11:47:37 -07:00
MediaPipe Team
51879ae81a Remove unnecessary architectures from image classifier ModelSpec
PiperOrigin-RevId: 481974529
2022-10-18 11:29:11 -07:00
MediaPipe Team
0428550d75 Open source model_maker/python/core/tasks and model_maker/python/vision/image_classifier
PiperOrigin-RevId: 481182271
2022-10-14 10:47:34 -07:00
MediaPipe Team
1b611c66bb Improve quantization support in model_maker/image_classifier
PiperOrigin-RevId: 480455944
2022-10-11 14:40:02 -07:00
MediaPipe Team
f4fd1063a7 Add helper methods to load saved model from external files in model maker.
PiperOrigin-RevId: 480444918
2022-10-11 13:58:05 -07:00
MediaPipe Team
6fa455f40e Dataclasses for text classifier
PiperOrigin-RevId: 480178697
2022-10-10 14:33:31 -07:00
MediaPipe Team
fb3b0e788e Open sourcing model_maker/core/util and model_maker/core/data
PiperOrigin-RevId: 478835650
2022-10-04 11:10:53 -07:00